diff --git a/tests/pipelines/chroma/test_pipeline_chroma.py b/tests/pipelines/chroma/test_pipeline_chroma.py index 9a2c850111..c47719d3e4 100644 --- a/tests/pipelines/chroma/test_pipeline_chroma.py +++ b/tests/pipelines/chroma/test_pipeline_chroma.py @@ -8,6 +8,7 @@ from diffusers import AutoencoderKL, ChromaPipeline, ChromaTransformer2DModel, F from diffusers.utils.testing_utils import torch_device from ..test_pipelines_common import ( + FluxIPAdapterTesterMixin, PipelineTesterMixin, check_qkv_fusion_matches_attn_procs_length, check_qkv_fusion_processors_exist,