Linoy Tsaban commited on
Commit
1fc5b9e
1 Parent(s): 48d3d17

Update inversion_utils.py

Browse files
Files changed (1) hide show
  1. 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, "").type(torch.float16)
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, "")
140
  timesteps = model.scheduler.timesteps.to(model.device)
141
  variance_noise_shape = (
142
  num_inference_steps,