FurnitureDemo

Running on Zero

blanchon commited on Jan 11

Commit

2e85b44

1 Parent(s): e4abc7b

Try to fix cuda error

Files changed (1) hide show

app.py CHANGED Viewed

@@ -128,7 +128,7 @@ def infer(
     results_images = pipe(
         prompt=[prompt] * batch_size,
         image=[image] * batch_size,
-        mask_image=[mask] * batch_size,
         height=FIXED_DIMENSION,
         width=FIXED_DIMENSION * 2,
         guidance_scale=guidance_scale,

     results_images = pipe(
         prompt=[prompt] * batch_size,
         image=[image] * batch_size,
+        mask_image=mask,
         height=FIXED_DIMENSION,
         width=FIXED_DIMENSION * 2,
         guidance_scale=guidance_scale,