niulx commited on
Commit
53382e5
1 Parent(s): a026553

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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", maximum=200, label="Batch size", interactive= True )
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 (