Skip to content

Commit 7787ec1

Browse files
committed
update
1 parent c5c7588 commit 7787ec1

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/diffusers/pipelines/flux/pipeline_flux.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -263,8 +263,7 @@ def _get_t5_prompt_embeds(
263263
)
264264

265265
prompt_embeds = self.text_encoder_2(
266-
text_input_ids.to(device),
267-
output_hidden_states=False,
266+
text_input_ids.to(device), output_hidden_states=False, attention_mask=text_inputs.attention_mask.to(device)
268267
)[0]
269268

270269
dtype = self.text_encoder_2.dtype

0 commit comments

Comments
 (0)