From 523150fb2c4c509ea4260348ba4f43343277fe1b Mon Sep 17 00:00:00 2001 From: BuildTools Date: Thu, 12 Jun 2025 21:47:35 -0600 Subject: [PATCH] fix import --- src/diffusers/pipelines/chroma/pipeline_chroma.py | 1 - .../utils/dummy_torch_and_transformers_objects.py | 15 +++++++++++++++ 2 files changed, 15 insertions(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/chroma/pipeline_chroma.py b/src/diffusers/pipelines/chroma/pipeline_chroma.py index b009554a47..f3074e0d09 100644 --- a/src/diffusers/pipelines/chroma/pipeline_chroma.py +++ b/src/diffusers/pipelines/chroma/pipeline_chroma.py @@ -182,7 +182,6 @@ class ChromaPipeline( transformer: ChromaTransformer2DModel, image_encoder: CLIPVisionModelWithProjection = None, feature_extractor: CLIPImageProcessor = None, - variant: str = "flux", ): super().__init__() diff --git a/src/diffusers/utils/dummy_torch_and_transformers_objects.py b/src/diffusers/utils/dummy_torch_and_transformers_objects.py index cc8f3e01ee..28d5b1beb5 100644 --- a/src/diffusers/utils/dummy_torch_and_transformers_objects.py +++ b/src/diffusers/utils/dummy_torch_and_transformers_objects.py @@ -287,6 +287,21 @@ class CLIPImageProjection(metaclass=DummyObject): requires_backends(cls, ["torch", "transformers"]) +class ChromaPipeline(metaclass=DummyObject): + _backends = ["torch", "transformers"] + + def __init__(self, *args, **kwargs): + requires_backends(self, ["torch", "transformers"]) + + @classmethod + def from_config(cls, *args, **kwargs): + requires_backends(cls, ["torch", "transformers"]) + + @classmethod + def from_pretrained(cls, *args, **kwargs): + requires_backends(cls, ["torch", "transformers"]) + + class CogVideoXFunControlPipeline(metaclass=DummyObject): _backends = ["torch", "transformers"]