Linoy Tsaban commited on
Commit
86ca2b5
·
1 Parent(s): 02f2630

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -105,8 +105,8 @@ def edit(input_image,
105
 
106
  cfg_scales_tensor = torch.Tensor(cfg_scales).view(-1,1,1,1).to(sd_pipe.device)
107
 
108
- text_embeddings = encode_text(model, prompts)
109
- uncond_embedding = encode_text(model, [""] * batch_size)
110
 
111
  if etas is None: etas = 0
112
  if type(etas) in [int, float]: etas = [etas]*sd_pipe.scheduler.num_inference_steps
 
105
 
106
  cfg_scales_tensor = torch.Tensor(cfg_scales).view(-1,1,1,1).to(sd_pipe.device)
107
 
108
+ text_embeddings = encode_text(sd_pipe, prompts)
109
+ uncond_embedding = encode_text(sd_pipe, [""] * batch_size)
110
 
111
  if etas is None: etas = 0
112
  if type(etas) in [int, float]: etas = [etas]*sd_pipe.scheduler.num_inference_steps