1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00

make fix-copies

This commit is contained in:
Aryan
2025-06-07 16:47:52 +02:00
parent a0617d504e
commit c2ab6c85ad
3 changed files with 30 additions and 1 deletions

View File

@@ -312,7 +312,6 @@ class CosmosTextToImagePipeline(DiffusionPipeline):
return prompt_embeds, negative_prompt_embeds
# Copied from diffusers.pipelines.cosmos.pipeline_cosmos_text2world.CosmosTextToWorldPipeline.prepare_latents
def prepare_latents(
self,
batch_size: int,

View File

@@ -1808,6 +1808,21 @@ class FlowMatchEulerDiscreteScheduler(metaclass=DummyObject):
requires_backends(cls, ["torch"])
class FlowMatchEulerEDMCosmos2_0Scheduler(metaclass=DummyObject):
_backends = ["torch"]
def __init__(self, *args, **kwargs):
requires_backends(self, ["torch"])
@classmethod
def from_config(cls, *args, **kwargs):
requires_backends(cls, ["torch"])
@classmethod
def from_pretrained(cls, *args, **kwargs):
requires_backends(cls, ["torch"])
class FlowMatchHeunDiscreteScheduler(metaclass=DummyObject):
_backends = ["torch"]

View File

@@ -407,6 +407,21 @@ class ConsisIDPipeline(metaclass=DummyObject):
requires_backends(cls, ["torch", "transformers"])
class CosmosTextToImagePipeline(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 CosmosTextToWorldPipeline(metaclass=DummyObject):
_backends = ["torch", "transformers"]