Disty0 commited on
Commit
b857280
·
1 Parent(s): d639c9b
Files changed (1) hide show
  1. pipeline.py +2 -2
pipeline.py CHANGED
@@ -367,14 +367,14 @@ class Zero123PlusPipeline(diffusers.StableDiffusionPipeline):
367
  global_embeds = global_embeds.unsqueeze(-2)
368
 
369
  if prompt_embeds is None:
370
- prompt_embeds, negative_prompt_embeds = self.encode_prompt(
371
  prompt,
372
  self.device,
373
  num_images_per_prompt,
374
  False,
375
  )
376
  ramp = global_embeds.new_tensor(self.config.ramping_coefficients).unsqueeze(-1)
377
- encoder_hidden_states = torch.cat([prompt_embeds, torch.zeros(prompt_embeds.shape, device=prompt_embeds.device)]) + global_embeds * ramp
378
  cak = dict(cond_lat=cond_lat)
379
  if hasattr(self.unet, "controlnet"):
380
  cak['control_depth'] = depth_image
 
367
  global_embeds = global_embeds.unsqueeze(-2)
368
 
369
  if prompt_embeds is None:
370
+ prompt_embeds, _ = self.encode_prompt(
371
  prompt,
372
  self.device,
373
  num_images_per_prompt,
374
  False,
375
  )
376
  ramp = global_embeds.new_tensor(self.config.ramping_coefficients).unsqueeze(-1)
377
+ encoder_hidden_states = prompt_embeds + global_embeds * ramp
378
  cak = dict(cond_lat=cond_lat)
379
  if hasattr(self.unet, "controlnet"):
380
  cak['control_depth'] = depth_image