eliphatfs commited on
Commit
983e66d
1 Parent(s): 5006f82

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +14 -6
pipeline.py CHANGED
@@ -361,12 +361,20 @@ class Zero123PlusPipeline(diffusers.StableDiffusionPipeline):
361
  global_embeds = encoded.image_embeds
362
  global_embeds = global_embeds.unsqueeze(-2)
363
 
364
- encoder_hidden_states = self._encode_prompt(
365
- prompt,
366
- self.device,
367
- num_images_per_prompt,
368
- False
369
- )
 
 
 
 
 
 
 
 
370
  ramp = global_embeds.new_tensor(self.config.ramping_coefficients).unsqueeze(-1)
371
  encoder_hidden_states = encoder_hidden_states + global_embeds * ramp
372
  cak = dict(cond_lat=cond_lat)
 
361
  global_embeds = encoded.image_embeds
362
  global_embeds = global_embeds.unsqueeze(-2)
363
 
364
+ if hasattr(self, "encode_prompt"):
365
+ encoder_hidden_states = self.encode_prompt(
366
+ prompt,
367
+ self.device,
368
+ num_images_per_prompt,
369
+ False
370
+ )[0]
371
+ else:
372
+ encoder_hidden_states = self._encode_prompt(
373
+ prompt,
374
+ self.device,
375
+ num_images_per_prompt,
376
+ False
377
+ )
378
  ramp = global_embeds.new_tensor(self.config.ramping_coefficients).unsqueeze(-1)
379
  encoder_hidden_states = encoder_hidden_states + global_embeds * ramp
380
  cak = dict(cond_lat=cond_lat)