Update model_index.json

model_index.json

@@ -1,11 +1,9 @@
 {
-  "_class_name": "StableDiffusionPipeline",
-  "vae": ["vae", null],
-  "text_encoder": ["text_encoder", null],
-  "tokenizer": ["tokenizer", null],
-  "unet": ["unet", null],
-  "scheduler": ["scheduler", null],
-  "safety_checker": ["safety_checker", null],
-  "feature_extractor": ["feature_extractor", null]
+    "_class_name": "StableDiffusionPipeline",
+    "vae": ["AutoencoderKL", null],
+    "text_encoder": ["CLIPTextModel", null],
+    "tokenizer": ["CLIPTokenizer", null],
+    "unet": ["UNet2DConditionModel", null],
+    "scheduler": ["DPMSolverMultistepScheduler", null]
 }