AlekseyCalvin commited on
Commit
6af5476
1 Parent(s): 00c43c3

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +3 -2
pipeline.py CHANGED
@@ -277,7 +277,6 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
277
  prompt=negative_prompt_2,
278
  device=device,
279
  num_images_per_prompt=num_images_per_prompt,
280
- max_sequence_length=max_sequence_length,
281
  )
282
 
283
  negative_clip_prompt_embed = torch.nn.functional.pad(
@@ -285,7 +284,9 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
285
  (0, t5_negative_prompt_embed.shape[-1] - negative_clip_prompt_embed.shape[-1]),
286
  )
287
 
288
- negative_prompt_embeds = torch.cat([negative_clip_prompt_embed, t5_negative_prompt_embed], dim=-2)
 
 
289
  negative_pooled_prompt_embeds = torch.cat(
290
  [negative_clip_prompt_embed, t5_negative_prompt_embed], dim=-1
291
  )
 
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(
 
284
  (0, t5_negative_prompt_embed.shape[-1] - negative_clip_prompt_embed.shape[-1]),
285
  )
286
 
287
+ negative_prompt_embeds = torch.cat([negative_clip_prompt_embed, t5_negative_prompt_embed], dim=-1)
288
+
289
+
290
  negative_pooled_prompt_embeds = torch.cat(
291
  [negative_clip_prompt_embed, t5_negative_prompt_embed], dim=-1
292
  )