model_index.json

{
  "_class_name": "sde_ve_pipeline/ScoreSdeVePipeline",
  "_module": "sde_ve_pipeline.py",
  "scheduler": [
    "sde_ve_scheduler",
    "ScoreSdeVeScheduler"
  ],
  "unet": [
    "unet_model",
    "UNet2DModel"
  ]
}