Spaces:
Running
on
Zero
Running
on
Zero
AlekseyCalvin
commited on
Commit
•
8447a86
1
Parent(s):
6af5476
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(
|