Skip to content

Commit 786babb

Browse files
committed
test_flux_prompt_embeds change not needed
1 parent 7db9b44 commit 786babb

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tests/pipelines/flux/test_pipeline_flux.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -138,13 +138,12 @@ def test_flux_prompt_embeds(self):
138138
inputs = self.get_dummy_inputs(torch_device)
139139
prompt = inputs.pop("prompt")
140140

141-
prompt_outputs = pipe.encode_prompt(
141+
(prompt_embeds, pooled_prompt_embeds, text_ids) = pipe.encode_prompt(
142142
prompt,
143143
prompt_2=None,
144144
device=torch_device,
145145
max_sequence_length=inputs["max_sequence_length"],
146146
)
147-
prompt_embeds, pooled_prompt_embeds = prompt_outputs[0], prompt_outputs[1]
148147
output_with_embeds = pipe(
149148
prompt_embeds=prompt_embeds,
150149
pooled_prompt_embeds=pooled_prompt_embeds,

0 commit comments

Comments
 (0)