Update pipeline.py
Browse files- pipeline.py +14 -6
pipeline.py
CHANGED
@@ -361,12 +361,20 @@ class Zero123PlusPipeline(diffusers.StableDiffusionPipeline):
|
|
361 |
global_embeds = encoded.image_embeds
|
362 |
global_embeds = global_embeds.unsqueeze(-2)
|
363 |
|
364 |
-
|
365 |
-
|
366 |
-
|
367 |
-
|
368 |
-
|
369 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
370 |
ramp = global_embeds.new_tensor(self.config.ramping_coefficients).unsqueeze(-1)
|
371 |
encoder_hidden_states = encoder_hidden_states + global_embeds * ramp
|
372 |
cak = dict(cond_lat=cond_lat)
|
|
|
361 |
global_embeds = encoded.image_embeds
|
362 |
global_embeds = global_embeds.unsqueeze(-2)
|
363 |
|
364 |
+
if hasattr(self, "encode_prompt"):
|
365 |
+
encoder_hidden_states = self.encode_prompt(
|
366 |
+
prompt,
|
367 |
+
self.device,
|
368 |
+
num_images_per_prompt,
|
369 |
+
False
|
370 |
+
)[0]
|
371 |
+
else:
|
372 |
+
encoder_hidden_states = self._encode_prompt(
|
373 |
+
prompt,
|
374 |
+
self.device,
|
375 |
+
num_images_per_prompt,
|
376 |
+
False
|
377 |
+
)
|
378 |
ramp = global_embeds.new_tensor(self.config.ramping_coefficients).unsqueeze(-1)
|
379 |
encoder_hidden_states = encoder_hidden_states + global_embeds * ramp
|
380 |
cak = dict(cond_lat=cond_lat)
|