Skip to content

Commit 4f1161d

Browse files
committed
make fix-copies
1 parent a63e543 commit 4f1161d

File tree

2 files changed

+15
-1
lines changed

2 files changed

+15
-1
lines changed

src/diffusers/pipelines/cosmos/pipeline_cosmos.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -173,7 +173,6 @@ def __init__(
173173
self.vae_scale_factor_spatial = self.vae.spatial_compression_ratio if getattr(self, "vae", None) else 8
174174
self.video_processor = VideoProcessor(vae_scale_factor=self.vae_scale_factor_spatial)
175175

176-
# Copied from diffusers.pipelines.mochi.pipeline_mochi.MochiPipeline._get_t5_prompt_embeds with 256->512
177176
def _get_t5_prompt_embeds(
178177
self,
179178
prompt: Union[str, List[str]] = None,

src/diffusers/utils/dummy_torch_and_transformers_objects.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -377,6 +377,21 @@ def from_pretrained(cls, *args, **kwargs):
377377
requires_backends(cls, ["torch", "transformers"])
378378

379379

380+
class CosmosPipeline(metaclass=DummyObject):
381+
_backends = ["torch", "transformers"]
382+
383+
def __init__(self, *args, **kwargs):
384+
requires_backends(self, ["torch", "transformers"])
385+
386+
@classmethod
387+
def from_config(cls, *args, **kwargs):
388+
requires_backends(cls, ["torch", "transformers"])
389+
390+
@classmethod
391+
def from_pretrained(cls, *args, **kwargs):
392+
requires_backends(cls, ["torch", "transformers"])
393+
394+
380395
class CycleDiffusionPipeline(metaclass=DummyObject):
381396
_backends = ["torch", "transformers"]
382397

0 commit comments

Comments
 (0)