Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 01:31:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

update cosmos

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-08-28 10:20:24 -04:00
parent ce5cdffacf
commit 9c1b6d3f08
2 changed files with 17 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/video_models/models_def.py
+14
View File
@@ -320,4 +320,18 @@ models = {
te_cls=transformers.T5EncoderModel,
dit_cls=diffusers.CogVideoXTransformer3DModel),
],
'nVidia Cosmos': [
Model(name='nvidia Cosmos Predict2 2B I2I',
url='https://huggingface.co/nvidia/Cosmos-Predict2-2B-Text2Image',
repo='nvidia/Cosmos-Predict2-2B-Video2World',
repo_cls=diffusers.Cosmos2VideoToWorldPipeline,
te_cls=transformers.T5EncoderModel,
dit_cls=diffusers.CosmosTransformer3DModel),
Model(name='nvidia Cosmos Predict2 2B I2I',
url='https://huggingface.co/nvidia/Cosmos-Predict2-2B-Text2Image',
repo='nvidia/Cosmos-Predict2-2B-Video2World',
repo_cls=diffusers.Cosmos2VideoToWorldPipeline,
te_cls=transformers.T5EncoderModel,
dit_cls=diffusers.CosmosTransformer3DModel),
],
}