Spaces:
Runtime error
Runtime error
Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
|
@@ -185,7 +185,7 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
|
|
| 185 |
prompt_embeds = prompt_embeds.pooler_output
|
| 186 |
prompt_embeds = prompt_embeds.to(dtype=self.text_encoder.dtype, device=device)
|
| 187 |
|
| 188 |
-
_, seq_len
|
| 189 |
|
| 190 |
# duplicate text embeddings for each generation per prompt, using mps friendly method
|
| 191 |
prompt_embeds = prompt_embeds.repeat(1, num_images_per_prompt, 1)
|
|
|
|
| 185 |
prompt_embeds = prompt_embeds.pooler_output
|
| 186 |
prompt_embeds = prompt_embeds.to(dtype=self.text_encoder.dtype, device=device)
|
| 187 |
|
| 188 |
+
_, seq_len = prompt_embeds.shape
|
| 189 |
|
| 190 |
# duplicate text embeddings for each generation per prompt, using mps friendly method
|
| 191 |
prompt_embeds = prompt_embeds.repeat(1, num_images_per_prompt, 1)
|