model_index.json

{
  "_class_name": "TextConditionalDDPMPipeline",
  "_diffusers_version": "0.32.2",
  "custom_pipeline": "models/text_diffusion_pipeline.py",
  "components": {
    "unet": { "type": "UNet2DConditionModel", "subfolder": "unet" },
    "text_encoder": { "type": "models.text_model.TransformerModel", "subfolder": "text_encoder" },
    "tokenizer": { "type": "Tokenizer", "file": "tokenizer.py" },
    "scheduler": { "type": "DDPMScheduler", "subfolder": "scheduler" }
  }
}