From 41be54eb01a9921ad5e35dd57edfc9f7f73cf234 Mon Sep 17 00:00:00 2001 From: CalamitousFelicitousness Date: Tue, 18 Aug 2026 00:32:44 +0100 Subject: [PATCH] fix(video): correct the cosmos row duplication and its model url The nVidia Cosmos engine carried two byte-identical Predict2 2B I2V rows, so the dropdown showed the name twice. The surviving row's url now points at the Video2World repo it loads; it pointed at the Text2Image image model. --- modules/video_models/models_def.py | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) 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',