1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

fix import

This commit is contained in:
BuildTools
2025-06-12 21:47:35 -06:00
parent 2bc51c8387
commit 523150fb2c
2 changed files with 15 additions and 1 deletions

View File

@@ -182,7 +182,6 @@ class ChromaPipeline(
transformer: ChromaTransformer2DModel,
image_encoder: CLIPVisionModelWithProjection = None,
feature_extractor: CLIPImageProcessor = None,
variant: str = "flux",
):
super().__init__()

View File

@@ -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"]