differential-diffusion

Configuration error

cocktailpeanut commited on Mar 1, 2024

Commit

cba9a1b

1 Parent(s): fdfa9a1

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -150,7 +150,7 @@ def inference(
     ).images
     base_device=None
     refiner_device = refiner.to(DEVICE)
-    edited_images = refiner_cuda(
         prompt=prompt,
         original_image=image,
         image=edited_images,

     ).images
     base_device=None
     refiner_device = refiner.to(DEVICE)
+    edited_images = refiner_device(
         prompt=prompt,
         original_image=image,
         image=edited_images,