Stable-Flow

Running on Zero

linoyts HF Staff commited on Feb 6

Commit

faedcba

verified ·

1 Parent(s): 6a4f2df

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -124,7 +124,7 @@ def invert_and_edit(image,
             inverted_latent_list=inverted_latent_list,
             mm_copy_blocks=MULTIMODAL_VITAL_LAYERS,
             single_copy_blocks=SINGLE_MODAL_VITAL_LAYERS,
-        ).images
     return output, inverted_latent_list.cpu(), do_inversion, seed

             inverted_latent_list=inverted_latent_list,
             mm_copy_blocks=MULTIMODAL_VITAL_LAYERS,
             single_copy_blocks=SINGLE_MODAL_VITAL_LAYERS,
+        ).images[1]
     return output, inverted_latent_list.cpu(), do_inversion, seed