Disty0
commited on
Commit
·
b857280
1
Parent(s):
d639c9b
Fix
Browse files- pipeline.py +2 -2
pipeline.py
CHANGED
@@ -367,14 +367,14 @@ class Zero123PlusPipeline(diffusers.StableDiffusionPipeline):
|
|
367 |
global_embeds = global_embeds.unsqueeze(-2)
|
368 |
|
369 |
if prompt_embeds is None:
|
370 |
-
prompt_embeds,
|
371 |
prompt,
|
372 |
self.device,
|
373 |
num_images_per_prompt,
|
374 |
False,
|
375 |
)
|
376 |
ramp = global_embeds.new_tensor(self.config.ramping_coefficients).unsqueeze(-1)
|
377 |
-
encoder_hidden_states =
|
378 |
cak = dict(cond_lat=cond_lat)
|
379 |
if hasattr(self.unet, "controlnet"):
|
380 |
cak['control_depth'] = depth_image
|
|
|
367 |
global_embeds = global_embeds.unsqueeze(-2)
|
368 |
|
369 |
if prompt_embeds is None:
|
370 |
+
prompt_embeds, _ = self.encode_prompt(
|
371 |
prompt,
|
372 |
self.device,
|
373 |
num_images_per_prompt,
|
374 |
False,
|
375 |
)
|
376 |
ramp = global_embeds.new_tensor(self.config.ramping_coefficients).unsqueeze(-1)
|
377 |
+
encoder_hidden_states = prompt_embeds + global_embeds * ramp
|
378 |
cak = dict(cond_lat=cond_lat)
|
379 |
if hasattr(self.unet, "controlnet"):
|
380 |
cak['control_depth'] = depth_image
|