dmt-entity / model_index.json
patrickvonplaten's picture
Fix deprecation warning by changing `CLIPFeatureExtractor` to `CLIPImageProcessor`.
a67af7e
raw
history blame
546 Bytes
{
"_class_name": "StableDiffusionPipeline",
"_diffusers_version": "0.4.0.dev0",
"feature_extractor": [
"transformers",
"CLIPImageProcessor"
],
"safety_checker": [
"stable_diffusion",
"StableDiffusionSafetyChecker"
],
"scheduler": [
"diffusers",
"PNDMScheduler"
],
"text_encoder": [
"transformers",
"CLIPTextModel"
],
"tokenizer": [
"transformers",
"CLIPTokenizer"
],
"unet": [
"diffusers",
"UNet2DConditionModel"
],
"vae": [
"diffusers",
"AutoencoderKL"
]
}