Update app.py

optimize lr and batchsize

app.py

@@ -202,11 +202,11 @@ with gr.Blocks() as demo:
                 with gr.Accordion(label="Advanced settings", open=False):
                     num_tokens = gr.Number(value="5", label="num tokens to represent each object", interactive= True)
                     num_tokens_global = num_tokens
-                    embedding_learning_rate = gr.Textbox(value="0.00005", label="Embedding optimization: Learning rate", interactive= True )
-                    max_emb_train_steps =  gr.Number(value="15", label="embedding optimization: Training steps", interactive= True )
+                    embedding_learning_rate = gr.Textbox(value="0.00025", label="Embedding optimization: Learning rate", interactive= True )
+                    max_emb_train_steps =  gr.Number(value="6", label="embedding optimization: Training steps", interactive= True )
                     
-                    diffusion_model_learning_rate = gr.Textbox(value="0.00002", label="UNet Optimization: Learning rate", interactive= True )
-                    max_diffusion_train_steps = gr.Number(value="10", label="UNet Optimization: Learning rate: Training steps", interactive= True )
+                    diffusion_model_learning_rate = gr.Textbox(value="0.0002", label="UNet Optimization: Learning rate", interactive= True )
+                    max_diffusion_train_steps = gr.Number(value="28", label="UNet Optimization: Learning rate: Training steps", interactive= True )
                     
                     train_batch_size = gr.Number(value="20", label="Batch size", interactive= True )
                     gradient_accumulation_steps=gr.Number(value="2", label="Gradient accumulation", interactive= True )
@@ -250,7 +250,7 @@ with gr.Blocks() as demo:
 
 
     #with gr.Tab(label="3 Editing"):
-        with gr.Tab(label="3.1 Text-based editing"):
+        with gr.Tab(label="Text-based editing"):
         
             with gr.Row():
                 with gr.Column():
@@ -263,8 +263,8 @@ with gr.Blocks() as demo:
                     with gr.Accordion(label="Advanced settings", open=False):
                         slider2 = gr.Slider(0, 20, step=1, label = 'mask id',  visible=False)
                         #tgt_index = gr.Number(value="0", label="Editing: Object index", interactive= True )
-                        guidance_scale = gr.Textbox(value="6", label="Editing: CFG guidance scale", interactive= True )
-                        num_sampling_steps = gr.Number(value="50", label="Editing: Sampling steps", interactive= True )
+                        guidance_scale = gr.Textbox(value="5", label="Editing: CFG guidance scale", interactive= True )
+                        num_sampling_steps = gr.Number(value="20", label="Editing: Sampling steps", interactive= True )
                         edge_thickness = gr.Number(value="10", label="Editing: Edge thickness", interactive= True )
                         strength = gr.Textbox(value="0.5", label="Editing: Mask strength", interactive= True )