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)
|