Spaces:
Runtime error
Runtime error
Commit
•
480e9b0
1
Parent(s):
ee0769d
Update app_training.py
Browse files- app_training.py +8 -5
app_training.py
CHANGED
@@ -25,17 +25,20 @@ def create_training_demo(trainer: Trainer,
|
|
25 |
max_lines=1,
|
26 |
placeholder='A man is surfing')
|
27 |
gr.Markdown('''
|
28 |
-
- Upload a video and write a prompt
|
29 |
''')
|
30 |
with gr.Box():
|
31 |
gr.Markdown('Output Model')
|
32 |
output_model_name = gr.Text(label='Name of your model',
|
|
|
33 |
max_lines=1)
|
34 |
delete_existing_model = gr.Checkbox(
|
35 |
label='Delete existing model of the same name',
|
36 |
-
value=False
|
37 |
-
|
38 |
-
|
|
|
|
|
39 |
|
40 |
with gr.Column():
|
41 |
with gr.Box():
|
@@ -74,7 +77,7 @@ def create_training_demo(trainer: Trainer,
|
|
74 |
value=100,
|
75 |
precision=0)
|
76 |
gr.Markdown('''
|
77 |
-
- The base model must be a
|
78 |
- Expected time to train a model for 300 steps: ~20 minutes with T4
|
79 |
- You can check the training status by pressing the "Open logs" button if you are running this on your Space.
|
80 |
''')
|
|
|
25 |
max_lines=1,
|
26 |
placeholder='A man is surfing')
|
27 |
gr.Markdown('''
|
28 |
+
- Upload a video and write a prompt that describes the video.
|
29 |
''')
|
30 |
with gr.Box():
|
31 |
gr.Markdown('Output Model')
|
32 |
output_model_name = gr.Text(label='Name of your model',
|
33 |
+
placeholder='The surfer man'
|
34 |
max_lines=1)
|
35 |
delete_existing_model = gr.Checkbox(
|
36 |
label='Delete existing model of the same name',
|
37 |
+
value=False)
|
38 |
+
validation_prompt = gr.Text(label='Validation Prompt', placeholder='a dog is surfing')
|
39 |
+
gr.Markdown('''
|
40 |
+
- The validation prompt should modify your `Training Prompt` in order to validate the model
|
41 |
+
''')
|
42 |
|
43 |
with gr.Column():
|
44 |
with gr.Box():
|
|
|
77 |
value=100,
|
78 |
precision=0)
|
79 |
gr.Markdown('''
|
80 |
+
- The base model must be a Stable Diffusion model compatible with [diffusers](https://github.com/huggingface/diffusers) library.
|
81 |
- Expected time to train a model for 300 steps: ~20 minutes with T4
|
82 |
- You can check the training status by pressing the "Open logs" button if you are running this on your Space.
|
83 |
''')
|