Disty0 commited on
Commit
cd79b31
·
1 Parent(s): 1e5ba03
Files changed (1) hide show
  1. pipeline.py +3 -3
pipeline.py CHANGED
@@ -332,7 +332,7 @@ class Zero123PlusPipeline(diffusers.StableDiffusionPipeline):
332
  self,
333
  image: Image.Image = None,
334
  prompt = "",
335
- negative_prompt = ""
336
  prompt_embeds = None,
337
  *args,
338
  num_images_per_prompt: Optional[int] = 1,
@@ -370,10 +370,10 @@ class Zero123PlusPipeline(diffusers.StableDiffusionPipeline):
370
  if prompt_embeds is None:
371
  prompt_embeds, negative_prompt_embeds = self.encode_prompt(
372
  prompt,
373
- negative_prompt=negative_prompt,
374
  self.device,
375
  num_images_per_prompt,
376
- False
 
377
  )
378
  encoder_hidden_states = torch.cat([prompt_embeds[1], negative_prompt_embeds[0]])
379
  ramp = global_embeds.new_tensor(self.config.ramping_coefficients).unsqueeze(-1)
 
332
  self,
333
  image: Image.Image = None,
334
  prompt = "",
335
+ negative_prompt = "",
336
  prompt_embeds = None,
337
  *args,
338
  num_images_per_prompt: Optional[int] = 1,
 
370
  if prompt_embeds is None:
371
  prompt_embeds, negative_prompt_embeds = self.encode_prompt(
372
  prompt,
 
373
  self.device,
374
  num_images_per_prompt,
375
+ False,
376
+ negative_prompt=negative_prompt,
377
  )
378
  encoder_hidden_states = torch.cat([prompt_embeds[1], negative_prompt_embeds[0]])
379
  ramp = global_embeds.new_tensor(self.config.ramping_coefficients).unsqueeze(-1)