modular_model_index.json

{
  "_blocks_class_name": "Cosmos3OmniBlocks",
  "_class_name": "Cosmos3OmniModularPipeline",
  "_diffusers_version": "0.39.0.dev0",
  "text_tokenizer": [
    "transformers",
    "Qwen2TokenizerFast",
    {
      "pretrained_model_name_or_path": "nvidia/Cosmos3-Super-Text2Image",
      "revision": null,
      "subfolder": "text_tokenizer",
      "type_hint": [
        "transformers",
        "Qwen2TokenizerFast"
      ],
      "variant": null
    }
  ],
  "vae": [
    "diffusers",
    "AutoencoderKLWan",
    {
      "pretrained_model_name_or_path": "nvidia/Cosmos3-Super-Text2Image",
      "revision": null,
      "subfolder": "vae",
      "type_hint": [
        "diffusers",
        "AutoencoderKLWan"
      ],
      "variant": null
    }
  ],
  "transformer": [
    "diffusers",
    "Cosmos3OmniTransformer",
    {
      "pretrained_model_name_or_path": "nvidia/Cosmos3-Super-Text2Image",
      "revision": null,
      "subfolder": "transformer",
      "type_hint": [
        "diffusers",
        "Cosmos3OmniTransformer"
      ],
      "variant": null
    }
  ],
  "scheduler": [
    "diffusers",
    "UniPCMultistepScheduler",
    {
      "pretrained_model_name_or_path": "nvidia/Cosmos3-Super-Text2Image",
      "revision": null,
      "subfolder": "scheduler",
      "type_hint": [
        "diffusers",
        "UniPCMultistepScheduler"
      ],
      "variant": null
    }
  ],
  "sound_tokenizer": [
    "diffusers",
    "Cosmos3AVAEAudioTokenizer",
    {
      "pretrained_model_name_or_path": "nvidia/Cosmos3-Super-Text2Image",
      "revision": null,
      "subfolder": "sound_tokenizer",
      "type_hint": [
        "diffusers",
        "Cosmos3AVAEAudioTokenizer"
      ],
      "variant": null
    }
  ]
}