modular_model_index.json

{
  "_blocks_class_name": "Cosmos3DistilledBlocks",
  "_class_name": "Cosmos3DistilledModularPipeline",
  "_diffusers_version": "0.40.0.dev0",
  "default_use_system_prompt": true,
  "distilled_sigmas": [
    1.0,
    0.9375,
    0.8333333333333334,
    0.625
  ],
  "enable_safety_checker": true,
  "is_distilled": true,
  "scheduler": [
    "diffusers",
    "FlowMatchEulerDiscreteScheduler",
    {
      "pretrained_model_name_or_path": "hf-internal-testing/tiny-cosmos3-distilled-modular-pipe",
      "revision": null,
      "subfolder": "scheduler",
      "type_hint": [
        "diffusers",
        "FlowMatchEulerDiscreteScheduler"
      ],
      "variant": null
    }
  ],
  "text_tokenizer": [
    "transformers",
    "Qwen2Tokenizer",
    {
      "pretrained_model_name_or_path": "hf-internal-testing/tiny-cosmos3-distilled-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-distilled-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-distilled-modular-pipe",
      "revision": null,
      "subfolder": "vae",
      "type_hint": [
        "diffusers",
        "AutoencoderKLWan"
      ],
      "variant": null
    }
  ]
}