diff --git a/modules/video_models/models_def.py b/modules/video_models/models_def.py index 12dcdc27d..b8872fdab 100644 --- a/modules/video_models/models_def.py +++ b/modules/video_models/models_def.py @@ -609,13 +609,7 @@ try: ], 'nVidia Cosmos': [ Model(name='nvidia Cosmos Predict2 2B I2V', - url='https://huggingface.co/nvidia/Cosmos-Predict2-2B-Text2Image', - repo='nvidia/Cosmos-Predict2-2B-Video2World', - repo_cls='Cosmos2VideoToWorldPipeline', - te_cls='T5EncoderModel', - dit_cls='CosmosTransformer3DModel'), - Model(name='nvidia Cosmos Predict2 2B I2V', - url='https://huggingface.co/nvidia/Cosmos-Predict2-2B-Text2Image', + url='https://huggingface.co/nvidia/Cosmos-Predict2-2B-Video2World', repo='nvidia/Cosmos-Predict2-2B-Video2World', repo_cls='Cosmos2VideoToWorldPipeline', te_cls='T5EncoderModel',