| { | |
| "_blocks_class_name": "Cosmos3OmniBlocks", | |
| "_class_name": "Cosmos3OmniModularPipeline", | |
| "_diffusers_version": "0.39.0", | |
| "is_distilled": false, | |
| "scheduler": [ | |
| "diffusers", | |
| "UniPCMultistepScheduler", | |
| { | |
| "pretrained_model_name_or_path": "Cosmos-Data/Cosmos-droid", | |
| "subfolder": "iter_002000/scheduler", | |
| "type_hint": [ | |
| "diffusers", | |
| "UniPCMultistepScheduler" | |
| ], | |
| "variant": null | |
| } | |
| ], | |
| "text_tokenizer": [ | |
| "transformers", | |
| "Qwen2TokenizerFast", | |
| { | |
| "pretrained_model_name_or_path": "Cosmos-Data/Cosmos-droid", | |
| "subfolder": "iter_002000/text_tokenizer", | |
| "type_hint": [ | |
| "transformers", | |
| "Qwen2TokenizerFast" | |
| ], | |
| "variant": null | |
| } | |
| ], | |
| "transformer": [ | |
| "diffusers", | |
| "Cosmos3OmniTransformer", | |
| { | |
| "pretrained_model_name_or_path": "Cosmos-Data/Cosmos-droid", | |
| "subfolder": "iter_002000/transformer", | |
| "type_hint": [ | |
| "diffusers", | |
| "Cosmos3OmniTransformer" | |
| ], | |
| "variant": null | |
| } | |
| ], | |
| "vae": [ | |
| "diffusers", | |
| "AutoencoderKLWan", | |
| { | |
| "pretrained_model_name_or_path": "Cosmos-Data/Cosmos-droid", | |
| "subfolder": "iter_002000/vae", | |
| "type_hint": [ | |
| "diffusers", | |
| "AutoencoderKLWan" | |
| ], | |
| "variant": null | |
| } | |
| ] | |
| } |