Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -220,12 +220,12 @@ with gr.Blocks() as demo:
|
|
220 |
num_tokens = gr.Number(value="5", label="num tokens to represent each object", interactive= True)
|
221 |
num_tokens_global = num_tokens
|
222 |
embedding_learning_rate = gr.Textbox(value="0.00025", label="Embedding optimization: Learning rate", interactive= True )
|
223 |
-
max_emb_train_steps = gr.Number(value="6", maximum=100, label="embedding optimization: Training steps", interactive= True )
|
224 |
|
225 |
diffusion_model_learning_rate = gr.Textbox(value="0.0002", label="UNet Optimization: Learning rate", interactive= True )
|
226 |
-
max_diffusion_train_steps = gr.Number(value="28", label="UNet Optimization: Learning rate: Training steps", interactive= True )
|
227 |
|
228 |
-
train_batch_size = gr.Number(value="20",
|
229 |
gradient_accumulation_steps=gr.Number(value="2", label="Gradient accumulation", interactive= True )
|
230 |
|
231 |
def run_optimization_wrapper (
|
|
|
220 |
num_tokens = gr.Number(value="5", label="num tokens to represent each object", interactive= True)
|
221 |
num_tokens_global = num_tokens
|
222 |
embedding_learning_rate = gr.Textbox(value="0.00025", label="Embedding optimization: Learning rate", interactive= True )
|
223 |
+
max_emb_train_steps = gr.Number(value="6", maximum="100", label="embedding optimization: Training steps", interactive= True )
|
224 |
|
225 |
diffusion_model_learning_rate = gr.Textbox(value="0.0002", label="UNet Optimization: Learning rate", interactive= True )
|
226 |
+
max_diffusion_train_steps = gr.Number(value="28", maximum="200", label="UNet Optimization: Learning rate: Training steps", interactive= True )
|
227 |
|
228 |
+
train_batch_size = gr.Number(value="20", label="Batch size", interactive= True )
|
229 |
gradient_accumulation_steps=gr.Number(value="2", label="Gradient accumulation", interactive= True )
|
230 |
|
231 |
def run_optimization_wrapper (
|