Update app.py

app.py

@@ -112,6 +112,8 @@ def auto_prompt_correction(prompt_ui,neg_prompt_ui,disable_auto_prompt_correctio
     
 def txt_to_img(prompt, neg_prompt, guidance, steps, width, height, generator):
     conditioning, pooled = compel([prompt, neg_prompt])
+    conditioning=conditioning.to("cuda")
+    pooled=pooled.to("cuda")
     
     result = pipe(
         prompt_embeds=conditioning[0:1],