model_index.json

{
  "_class_name": "PartragPipeline",
  "_diffusers_version": "0.35.1",
  "feature_extractor_dinov2": [
    "transformers",
    "BitImageProcessor"
  ],
  "image_encoder_dinov2": [
    "transformers",
    "Dinov2Model"
  ],
  "scheduler": [
    "src.schedulers.scheduling_rectified_flow",
    "RectifiedFlowScheduler"
  ],
  "transformer": [
    "src.models.transformers.partrag_transformer",
    "PartragDiTModel"
  ],
  "vae": [
    "src.models.autoencoders.autoencoder_kl_triposg",
    "TripoSGVAEModel"
  ]
}