Spaces:
Runtime error
Runtime error
Update pipeline.py
Browse files- pipeline.py +2 -0
pipeline.py
CHANGED
|
@@ -154,6 +154,7 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
|
|
| 154 |
self,
|
| 155 |
prompt: Union[str, List[str]],
|
| 156 |
num_images_per_prompt: int = 1,
|
|
|
|
| 157 |
device: Optional[torch.device] = None,
|
| 158 |
):
|
| 159 |
device = device or self._execution_device
|
|
@@ -277,6 +278,7 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
|
|
| 277 |
prompt=negative_prompt_2,
|
| 278 |
device=device,
|
| 279 |
num_images_per_prompt=num_images_per_prompt,
|
|
|
|
| 280 |
)
|
| 281 |
|
| 282 |
negative_clip_prompt_embed = torch.nn.functional.pad(
|
|
|
|
| 154 |
self,
|
| 155 |
prompt: Union[str, List[str]],
|
| 156 |
num_images_per_prompt: int = 1,
|
| 157 |
+
max_sequence_length: int = 512,
|
| 158 |
device: Optional[torch.device] = None,
|
| 159 |
):
|
| 160 |
device = device or self._execution_device
|
|
|
|
| 278 |
prompt=negative_prompt_2,
|
| 279 |
device=device,
|
| 280 |
num_images_per_prompt=num_images_per_prompt,
|
| 281 |
+
max_sequence_length=max_sequence_length,
|
| 282 |
)
|
| 283 |
|
| 284 |
negative_clip_prompt_embed = torch.nn.functional.pad(
|