LibreFLUX_LoRAs_Gallery

Running on Zero

AlekseyCalvin commited on Oct 19, 2024

Commit

f20a809

verified ·

1 Parent(s): f5d226e

Update pipeline.py

Files changed (1) hide show

pipeline.py CHANGED Viewed

@@ -236,6 +236,7 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
                 prompt=prompt,
                 device=device,
                 num_images_per_prompt=num_images_per_prompt,
             )
             prompt_embeds = self._get_t5_prompt_embeds(
                 prompt=prompt_2,
@@ -270,6 +271,7 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
                 prompt=negative_prompt,
                 device=device,
                 num_images_per_prompt=num_images_per_prompt,
             )
             negative_prompt_embeds = self._get_t5_prompt_embeds(

                 prompt=prompt,
                 device=device,
                 num_images_per_prompt=num_images_per_prompt,
+                max_sequence_length=max_sequence_length,
             )
             prompt_embeds = self._get_t5_prompt_embeds(
                 prompt=prompt_2,
                 prompt=negative_prompt,
                 device=device,
                 num_images_per_prompt=num_images_per_prompt,
+                max_sequence_length=max_sequence_length,
             )
             negative_prompt_embeds = self._get_t5_prompt_embeds(