Upload app.py

app.py

@@ -114,8 +114,8 @@ def update_selection(evt: gr.SelectData, width, height):
 @spaces.GPU(duration=70)
 def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, lora_scale, cn_on, progress=gr.Progress(track_tqdm=True)):
     pipe.to("cuda")
-    controlnet.to("cuda")
-    controlnet_union.to("cuda")
+    #controlnet.to("cuda")
+    #controlnet_union.to("cuda")
     generator = torch.Generator(device="cuda").manual_seed(seed)
     
     with calculateDuration("Generating image"):