model_index.json

{
  "_class_name": "SpatialGenDiffusionPipeline",
  "_diffusers_version": "0.32.0",
  "depth_vae": [
    "diffusers",
    "AutoencoderTiny"
  ],
  "feature_extractor": [
    null,
    null
  ],
  "ray_encoder": [
    "src.models.pose_adapter",
    "RayMapEncoder"
  ],
  "requires_safety_checker": true,
  "safety_checker": [
    null,
    null
  ],
  "scheduler": [
    "diffusers",
    "DDPMScheduler"
  ],
  "text_encoder": [
    "transformers",
    "CLIPTextModel"
  ],
  "tokenizer": [
    "transformers",
    "CLIPTokenizer"
  ],
  "unet": [
    "diffusers_spatialgen.models.unets.unet_mvmm2d_condition",
    "UNetMVMM2DConditionModel"
  ],
  "vae": [
    "diffusers",
    "AutoencoderKL"
  ]
}