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