Disty0 commited on
Commit
56520d0
·
1 Parent(s): a0f3eec
Files changed (1) hide show
  1. pipeline.py +3 -1
pipeline.py CHANGED
@@ -366,7 +366,7 @@ class Zero123PlusPipeline(diffusers.StableDiffusionPipeline):
366
  global_embeds = encoded.image_embeds
367
  global_embeds = global_embeds.unsqueeze(-2)
368
 
369
- if prompt_embeds is not None:
370
  encoder_hidden_states = self._encode_prompt(
371
  prompts,
372
  self.device,
@@ -375,6 +375,8 @@ class Zero123PlusPipeline(diffusers.StableDiffusionPipeline):
375
  )
376
  ramp = global_embeds.new_tensor(self.config.ramping_coefficients).unsqueeze(-1)
377
  encoder_hidden_states = encoder_hidden_states + global_embeds * ramp
 
 
378
  cak = dict(cond_lat=cond_lat)
379
  if hasattr(self.unet, "controlnet"):
380
  cak['control_depth'] = depth_image
 
366
  global_embeds = encoded.image_embeds
367
  global_embeds = global_embeds.unsqueeze(-2)
368
 
369
+ if prompt_embeds is None:
370
  encoder_hidden_states = self._encode_prompt(
371
  prompts,
372
  self.device,
 
375
  )
376
  ramp = global_embeds.new_tensor(self.config.ramping_coefficients).unsqueeze(-1)
377
  encoder_hidden_states = encoder_hidden_states + global_embeds * ramp
378
+ else:
379
+ encoder_hidden_states = prompt_embeds
380
  cak = dict(cond_lat=cond_lat)
381
  if hasattr(self.unet, "controlnet"):
382
  cak['control_depth'] = depth_image