Disty0 commited on
Commit ·
56520d0
1
Parent(s): a0f3eec
- 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
|
| 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
|