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, no_cfg_until_timestep, lora_scale, progress):
+def generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height, negative_prompt, lora_scale, progress, no_cfg_until_timestep):
     pipe.to("cuda")
     generator = torch.Generator(device="cuda").manual_seed(seed)
     
@@ -118,12 +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},
+            no_cfg_until_timestep=2,
         ).images[0]
     return image
 
-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)):
+def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, negative_prompt, lora_scale, no_cfg_until_timestep=2, progress=gr.Progress(track_tqdm=True)):
     if selected_index is None:
         raise gr.Error("You must select a LoRA before proceeding.")
 
@@ -153,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, no_cfg_until_timestep, lora_scale, progress)
+    image = generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height, negative_prompt, lora_scale, progress, no_cfg_until_timestep)
     pipe.to("cpu")
     pipe.unload_lora_weights()
     return image, seed