Linoy Tsaban
commited on
Commit
•
48d3d17
1
Parent(s):
002a360
Update inversion_utils.py
Browse files- inversion_utils.py +1 -1
inversion_utils.py
CHANGED
@@ -136,7 +136,7 @@ def inversion_forward_process(model, x0,
|
|
136 |
|
137 |
if not prompt=="":
|
138 |
text_embeddings = encode_text(model, prompt)
|
139 |
-
uncond_embedding = encode_text(model, "")
|
140 |
timesteps = model.scheduler.timesteps.to(model.device)
|
141 |
variance_noise_shape = (
|
142 |
num_inference_steps,
|
|
|
136 |
|
137 |
if not prompt=="":
|
138 |
text_embeddings = encode_text(model, prompt)
|
139 |
+
uncond_embedding = encode_text(model, "").type(torch.float16)
|
140 |
timesteps = model.scheduler.timesteps.to(model.device)
|
141 |
variance_noise_shape = (
|
142 |
num_inference_steps,
|