Spaces:

iitolstykh
/

VIBE-Image-Edit-DEMO

Running on Zero

iitolstykh commited on Dec 31, 2025

Commit

c525562

verified ·

1 Parent(s): 83c07d3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -69,7 +69,10 @@ def generate_img(
     seed,
     pipeline: ImageEditor,
 ):
-    pipeline.pipe.to("cuda").to(pipeline.weight_dtype)
     pil_image = PILImage.fromarray(np_image)

     seed,
     pipeline: ImageEditor,
 ):
+    # FIXME
+    pipeline = load_pipeline()
+    # pipeline.pipe.to("cuda").to(pipeline.weight_dtype)
     pil_image = PILImage.fromarray(np_image)