Spaces:

editing-images
/

ledits

Running on A10G

Linoy Tsaban commited on May 25, 2023

Commit

002a360

•

1 Parent(s): e19684c

Update inversion_utils.py

Files changed (1) hide show

inversion_utils.py CHANGED Viewed

@@ -136,7 +136,7 @@ def inversion_forward_process(model, x0,
     if not prompt=="":
         text_embeddings = encode_text(model, prompt)
-    uncond_embedding = encode_text(model, "")
     timesteps = model.scheduler.timesteps.to(model.device)
     variance_noise_shape = (
         num_inference_steps,

     if not prompt=="":
         text_embeddings = encode_text(model, prompt)
+    uncond_embedding = encode_text(model, "")).type(torch.float16)
     timesteps = model.scheduler.timesteps.to(model.device)
     variance_noise_shape = (
         num_inference_steps,