model_index.json

{
  "_class_name": "RGBXVideoDiffusionPipeline",
  "_diffusers_version": "0.28.0",
  "env_encoder": [
    "src.models.env_encoder",
    "EnvEncoder"
  ],
  "feature_extractor": [
    null,
    null
  ],
  "image_encoder": [
    null,
    null
  ],
  "scheduler": [
    "diffusers",
    "EulerDiscreteScheduler"
  ],
  "unet": [
    "src.models.custom_unet_st",
    "UNetCustomSpatioTemporalConditionModel"
  ],
  "vae": [
    "diffusers",
    "AutoencoderKLTemporalDecoder"
  ]
}