AlekseyCalvin commited on
Commit
8447a86
1 Parent(s): 6af5476

Update pipeline.py

Browse files
Files changed (1) hide show
  1. 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(