Skip to content

Commit 735fb0e

Browse files
make style + make fix-copies
1 parent b76f9f2 commit 735fb0e

File tree

3 files changed

+62
-3
lines changed

3 files changed

+62
-3
lines changed

src/diffusers/pipelines/cosmos/pipeline_cosmos2_5_predict.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -233,7 +233,6 @@ def __init__(
233233
if self.latents_mean is None or self.latents_std is None:
234234
raise ValueError("VAE configuration must define both `latents_mean` and `latents_std`.")
235235

236-
# Copied from diffusers.pipelines.cosmos.pipeline_cosmos_text2world.CosmosTextToWorldPipeline._get_prompt_embeds
237236
def _get_prompt_embeds(
238237
self,
239238
prompt: Union[str, List[str]] = None,
@@ -384,7 +383,7 @@ def encode_prompt(
384383

385384
return prompt_embeds, negative_prompt_embeds
386385

387-
# Modified from diffusers.pipelines.cosmos.pipeline_cosmos2_video2world.Cosmos2VideoToWorldPipeline.prepare_latents and
386+
# Modified from diffusers.pipelines.cosmos.pipeline_cosmos2_video2world.Cosmos2VideoToWorldPipeline.prepare_latents and
388387
# diffusers.pipelines.cosmos.pipeline_cosmos2_video2world.Cosmos2TextToImagePipeline.prepare_latents
389388
def prepare_latents(
390389
self,

src/diffusers/utils/dummy_torch_and_transformers_objects.py

Lines changed: 60 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -767,6 +767,66 @@ def from_pretrained(cls, *args, **kwargs):
767767
requires_backends(cls, ["torch", "transformers"])
768768

769769

770+
class Cosmos2_5_PredictBasePipeline(metaclass=DummyObject):
771+
_backends = ["torch", "transformers"]
772+
773+
def __init__(self, *args, **kwargs):
774+
requires_backends(self, ["torch", "transformers"])
775+
776+
@classmethod
777+
def from_config(cls, *args, **kwargs):
778+
requires_backends(cls, ["torch", "transformers"])
779+
780+
@classmethod
781+
def from_pretrained(cls, *args, **kwargs):
782+
requires_backends(cls, ["torch", "transformers"])
783+
784+
785+
class Cosmos2_5_PredictImage2World(metaclass=DummyObject):
786+
_backends = ["torch", "transformers"]
787+
788+
def __init__(self, *args, **kwargs):
789+
requires_backends(self, ["torch", "transformers"])
790+
791+
@classmethod
792+
def from_config(cls, *args, **kwargs):
793+
requires_backends(cls, ["torch", "transformers"])
794+
795+
@classmethod
796+
def from_pretrained(cls, *args, **kwargs):
797+
requires_backends(cls, ["torch", "transformers"])
798+
799+
800+
class Cosmos2_5_PredictText2World(metaclass=DummyObject):
801+
_backends = ["torch", "transformers"]
802+
803+
def __init__(self, *args, **kwargs):
804+
requires_backends(self, ["torch", "transformers"])
805+
806+
@classmethod
807+
def from_config(cls, *args, **kwargs):
808+
requires_backends(cls, ["torch", "transformers"])
809+
810+
@classmethod
811+
def from_pretrained(cls, *args, **kwargs):
812+
requires_backends(cls, ["torch", "transformers"])
813+
814+
815+
class Cosmos2_5_PredictVideo2World(metaclass=DummyObject):
816+
_backends = ["torch", "transformers"]
817+
818+
def __init__(self, *args, **kwargs):
819+
requires_backends(self, ["torch", "transformers"])
820+
821+
@classmethod
822+
def from_config(cls, *args, **kwargs):
823+
requires_backends(cls, ["torch", "transformers"])
824+
825+
@classmethod
826+
def from_pretrained(cls, *args, **kwargs):
827+
requires_backends(cls, ["torch", "transformers"])
828+
829+
770830
class Cosmos2TextToImagePipeline(metaclass=DummyObject):
771831
_backends = ["torch", "transformers"]
772832

tests/schedulers/test_scheduler_unipc.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -414,7 +414,7 @@ def test_flow_and_karras_sigmas_values(self):
414414
num_train_timesteps=num_train_timesteps,
415415
)
416416
scheduler.set_timesteps(num_inference_steps=num_inference_steps)
417-
417+
418418
expected_sigmas = [
419419
0.9950248599052429,
420420
0.9787454605102539,

0 commit comments

Comments
 (0)