Spaces:
Running
on
Zero
Running
on
Zero
HuiZhang
commited on
Update src/pipeline/pipeline_CreatiLayout.py
Browse files
src/pipeline/pipeline_CreatiLayout.py
CHANGED
|
@@ -299,7 +299,7 @@ class CreatiLayoutSD3Pipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingleF
|
|
| 299 |
pooled_prompt_embeds = prompt_embeds[0]
|
| 300 |
|
| 301 |
if clip_skip is None:
|
| 302 |
-
prompt_embeds = prompt_embeds.hidden_states[-2]
|
| 303 |
else:
|
| 304 |
prompt_embeds = prompt_embeds.hidden_states[-(clip_skip + 2)]
|
| 305 |
|
|
@@ -313,7 +313,7 @@ class CreatiLayoutSD3Pipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingleF
|
|
| 313 |
pooled_prompt_embeds = pooled_prompt_embeds.repeat(1, num_images_per_prompt, 1)
|
| 314 |
pooled_prompt_embeds = pooled_prompt_embeds.view(batch_size * num_images_per_prompt, -1)
|
| 315 |
|
| 316 |
-
return prompt_embeds, pooled_prompt_embeds
|
| 317 |
|
| 318 |
def encode_prompt(
|
| 319 |
self,
|
|
|
|
| 299 |
pooled_prompt_embeds = prompt_embeds[0]
|
| 300 |
|
| 301 |
if clip_skip is None:
|
| 302 |
+
prompt_embeds = prompt_embeds.hidden_states[-2]
|
| 303 |
else:
|
| 304 |
prompt_embeds = prompt_embeds.hidden_states[-(clip_skip + 2)]
|
| 305 |
|
|
|
|
| 313 |
pooled_prompt_embeds = pooled_prompt_embeds.repeat(1, num_images_per_prompt, 1)
|
| 314 |
pooled_prompt_embeds = pooled_prompt_embeds.view(batch_size * num_images_per_prompt, -1)
|
| 315 |
|
| 316 |
+
return prompt_embeds, pooled_prompt_embeds
|
| 317 |
|
| 318 |
def encode_prompt(
|
| 319 |
self,
|