Skip to content

Commit 9a2ec46

Browse files
committed
fixes
1 parent f43dd95 commit 9a2ec46

File tree

2 files changed

+0
-3
lines changed

2 files changed

+0
-3
lines changed

tests/pipelines/flux/test_pipeline_flux.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,6 @@ class FluxPipelineFastTests(
3131
pipeline_class = FluxPipeline
3232
params = frozenset(["prompt", "height", "width", "guidance_scale", "prompt_embeds", "pooled_prompt_embeds"])
3333
batch_params = frozenset(["prompt"])
34-
prompt_embed_kwargs = ("prompt_embeds", "pooled_prompt_embeds", "text_ids")
3534

3635
# there is no xformers processor for Flux
3736
test_xformers_attention = False

tests/pipelines/test_pipelines_common.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -993,8 +993,6 @@ class PipelineTesterMixin:
993993
test_layerwise_casting = False
994994
supports_dduf = True
995995

996-
prompt_embed_kwargs = ("prompt_embeds", "negative_prompt_embeds") # most common return-type across the pipelines.
997-
998996
def get_generator(self, seed):
999997
device = torch_device if torch_device != "mps" else "cpu"
1000998
generator = torch.Generator(device).manual_seed(seed)

0 commit comments

Comments
 (0)