Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -209,7 +209,7 @@ with gr.Blocks() as demo:
|
|
209 |
diffusion_model_learning_rate = gr.Textbox(value="0.00002", label="UNet Optimization: Learning rate", interactive= True )
|
210 |
max_diffusion_train_steps = gr.Number(value="10", label="UNet Optimization: Learning rate: Training steps", interactive= True )
|
211 |
|
212 |
-
train_batch_size = gr.Number(value="
|
213 |
gradient_accumulation_steps=gr.Number(value="2", label="Gradient accumulation", interactive= True )
|
214 |
|
215 |
add_button = gr.Button("Run optimization")
|
|
|
209 |
diffusion_model_learning_rate = gr.Textbox(value="0.00002", label="UNet Optimization: Learning rate", interactive= True )
|
210 |
max_diffusion_train_steps = gr.Number(value="10", label="UNet Optimization: Learning rate: Training steps", interactive= True )
|
211 |
|
212 |
+
train_batch_size = gr.Number(value="20", label="Batch size", interactive= True )
|
213 |
gradient_accumulation_steps=gr.Number(value="2", label="Gradient accumulation", interactive= True )
|
214 |
|
215 |
add_button = gr.Button("Run optimization")
|