uuu

app.py

@@ -67,18 +67,17 @@ pipe.init_ipadapter(
 def gui_generation(prompt, ref_img, guidance_scale, ipadapter_scale):
 
     # Ensure the pipeline runs with correct dtype and device
-    with torch.autocast("cuda", dtype=torch.bfloat16):
-        image = pipe(
-            width=1024,
-            height=1024,
-            prompt=prompt,
-            negative_prompt="lowres, low quality, worst quality",
-            num_inference_steps=24,
-            guidance_scale=guidance_scale,
-            generator=torch.Generator("cuda").manual_seed(42),
-            clip_image=ref_img.convert('RGB'),
-            ipadapter_scale=ipadapter_scale,
-        ).images[0]
+    image = pipe(
+        width=1024,
+        height=1024,
+        prompt=prompt,
+        negative_prompt="lowres, low quality, worst quality",
+        num_inference_steps=24,
+        guidance_scale=guidance_scale,
+        generator=torch.Generator("cuda").manual_seed(42),
+        clip_image=ref_img.convert('RGB'),
+        ipadapter_scale=ipadapter_scale,
+    ).images[0]
 
     return image