Disty0
commited on
Commit
·
cd79b31
1
Parent(s):
1e5ba03
fix typo
Browse files- 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)
|