model_index.json

{
  "_class_name": "DifixPipeline",
  "_diffusers_version": "0.25.1",
  "_name_or_path": "nvidia/difix_ref",
  "feature_extractor": [
    null,
    null
  ],
  "image_encoder": [
    null,
    null
  ],
  "requires_safety_checker": true,
  "safety_checker": [
    null,
    null
  ],
  "scheduler": [
    "diffusers",
    "DDPMScheduler"
  ],
  "text_encoder": [
    "transformers",
    "CLIPTextModel"
  ],
  "tokenizer": [
    "transformers",
    "CLIPTokenizer"
  ],
  "unet": [
    "unet_2d_condition",
    "UNet2DConditionModel"
  ],
  "vae": [
    "autoencoder_kl",
    "AutoencoderKL"
  ]
}