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.
This commit is contained in:
CalamitousFelicitousness
2026-08-18 00:32:44 +01:00
parent e23a70af34
commit 41be54eb01
+1 -7
View File
@@ -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',