niulx commited on
Commit
5c1d9eb
1 Parent(s): ddd0257

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -205,10 +205,10 @@ with gr.Blocks() as demo:
205
  num_tokens = gr.Number(value="5", label="num tokens to represent each object", interactive= True)
206
  num_tokens_global = num_tokens
207
  embedding_learning_rate = gr.Textbox(value="0.00005", label="Embedding optimization: Learning rate", interactive= True )
208
- max_emb_train_steps = gr.Number(value="100", label="embedding optimization: Training steps", interactive= True )
209
 
210
  diffusion_model_learning_rate = gr.Textbox(value="0.00002", label="UNet Optimization: Learning rate", interactive= True )
211
- max_diffusion_train_steps = gr.Number(value="100", label="UNet Optimization: Learning rate: Training steps", interactive= True )
212
 
213
  train_batch_size = gr.Number(value="5", label="Batch size", interactive= True )
214
  gradient_accumulation_steps=gr.Number(value="5", label="Gradient accumulation", interactive= True )
 
205
  num_tokens = gr.Number(value="5", label="num tokens to represent each object", interactive= True)
206
  num_tokens_global = num_tokens
207
  embedding_learning_rate = gr.Textbox(value="0.00005", label="Embedding optimization: Learning rate", interactive= True )
208
+ max_emb_train_steps = gr.Number(value="80", label="embedding optimization: Training steps", interactive= True )
209
 
210
  diffusion_model_learning_rate = gr.Textbox(value="0.00002", label="UNet Optimization: Learning rate", interactive= True )
211
+ max_diffusion_train_steps = gr.Number(value="80", label="UNet Optimization: Learning rate: Training steps", interactive= True )
212
 
213
  train_batch_size = gr.Number(value="5", label="Batch size", interactive= True )
214
  gradient_accumulation_steps=gr.Number(value="5", label="Gradient accumulation", interactive= True )