Update app.py

app.py

@@ -104,7 +104,7 @@ def update_selection(evt: gr.SelectData, width, height):
     )
 
 @spaces.GPU(duration=70)
-def generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height, negative_prompt, lora_scale, progress):
+def generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height, negative_prompt, no_cfg_until_timestep, lora_scale, progress):
     pipe.to("cuda")
     generator = torch.Generator(device="cuda").manual_seed(seed)
     
@@ -118,11 +118,12 @@ def generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height,
             height=height,
             generator=generator,
             negative_prompt=negative_prompt,
+            no_cfg_until_timestep=2,
             joint_attention_kwargs={"scale": lora_scale},
         ).images[0]
     return image
 
-def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, negative_prompt, lora_scale, progress=gr.Progress(track_tqdm=True)):
+def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, negative_prompt, no_cfg_until_timestep=2, lora_scale, progress=gr.Progress(track_tqdm=True)):
     if selected_index is None:
         raise gr.Error("You must select a LoRA before proceeding.")
 
@@ -152,7 +153,7 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
         if randomize_seed:
             seed = random.randint(0, MAX_SEED)
     
-    image = generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height, negative_prompt, lora_scale, progress)
+    image = generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height, negative_prompt, no_cfg_until_timestep, lora_scale, progress)
     pipe.to("cpu")
     pipe.unload_lora_weights()
     return image, seed  
@@ -206,8 +207,8 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
         with gr.Accordion("Advanced Settings", open=True):
             with gr.Column():
                 with gr.Row():
-                    cfg_scale = gr.Slider(label="CFG Scale", minimum=0, maximum=20, step=0.5, value=3.0)
-                    steps = gr.Slider(label="Steps", minimum=1, maximum=50, step=1, value=12)
+                    cfg_scale = gr.Slider(label="CFG Scale", minimum=0, maximum=20, step=0.5, value=2.5)
+                    steps = gr.Slider(label="Steps", minimum=1, maximum=50, step=1, value=20)
                 
                 with gr.Row():
                     width = gr.Slider(label="Width", minimum=256, maximum=1536, step=64, value=1024)
@@ -216,7 +217,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
                 with gr.Row():
                     randomize_seed = gr.Checkbox(True, label="Randomize seed")
                     seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0, randomize=True)
-                    lora_scale = gr.Slider(label="LoRA Scale", minimum=0, maximum=2.0, step=0.01, value=1.05)
+                    lora_scale = gr.Slider(label="LoRA Scale", minimum=0, maximum=2.0, step=0.01, value=0.8)
 
     gallery.select(
         update_selection,