modular_model_index.json

{
  "_blocks_class_name": "Cosmos3OmniBlocks",
  "_class_name": "Cosmos3OmniModularPipeline",
  "_diffusers_version": "0.40.0.dev0",
  "default_use_system_prompt": true,
  "enable_safety_checker": true,
  "scheduler": [
    "diffusers",
    "UniPCMultistepScheduler",
    {
      "pretrained_model_name_or_path": "hf-internal-testing/tiny-cosmos3-modular-pipe",
      "revision": null,
      "subfolder": "scheduler",
      "type_hint": [
        "diffusers",
        "UniPCMultistepScheduler"
      ],
      "variant": null
    }
  ],
  "sound_tokenizer": [
    "diffusers",
    "Cosmos3AVAEAudioTokenizer",
    {
      "pretrained_model_name_or_path": "hf-internal-testing/tiny-cosmos3-modular-pipe",
      "revision": null,
      "subfolder": "sound_tokenizer",
      "type_hint": [
        "diffusers",
        "Cosmos3AVAEAudioTokenizer"
      ],
      "variant": null
    }
  ],
  "text_tokenizer": [
    "transformers",
    "Qwen2Tokenizer",
    {
      "pretrained_model_name_or_path": "hf-internal-testing/tiny-cosmos3-modular-pipe",
      "revision": null,
      "subfolder": "text_tokenizer",
      "type_hint": [
        "transformers",
        "Qwen2Tokenizer"
      ],
      "variant": null
    }
  ],
  "transformer": [
    "diffusers",
    "Cosmos3OmniTransformer",
    {
      "pretrained_model_name_or_path": "hf-internal-testing/tiny-cosmos3-modular-pipe",
      "revision": null,
      "subfolder": "transformer",
      "type_hint": [
        "diffusers",
        "Cosmos3OmniTransformer"
      ],
      "variant": null
    }
  ],
  "use_native_flow_schedule": false,
  "vae": [
    "diffusers",
    "AutoencoderKLWan",
    {
      "pretrained_model_name_or_path": "hf-internal-testing/tiny-cosmos3-modular-pipe",
      "revision": null,
      "subfolder": "vae",
      "type_hint": [
        "diffusers",
        "AutoencoderKLWan"
      ],
      "variant": null
    }
  ]
}