Spaces:

linoyts
/

edit_friendly_ddpm_inversion

Running on A10G

Linoy Tsaban commited on Jun 1, 2023

Commit

b2af899

•

1 Parent(s): 86ca2b5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -147,8 +147,8 @@ def edit(input_image,
             x0_dec = sd_pipe.vae.decode(1 / 0.18215 * xt).sample
             if x0_dec.dim()<4:
                 x0_dec = x0_dec[None,:,:,:]
-                interm_img = image_grid(x0_dec)
-                yield interm_img
     yield interm_img

             x0_dec = sd_pipe.vae.decode(1 / 0.18215 * xt).sample
             if x0_dec.dim()<4:
                 x0_dec = x0_dec[None,:,:,:]
+            interm_img = image_grid(x0_dec)
+            yield interm_img
     yield interm_img