Update model_index.json

model_index.json

@@ -1,18 +1,29 @@
 {
+  "_class_name": "StableDiffusionPipeline",
   "model_type": "stable-diffusion",
-  "auto_encoder": {
+  "unet": {
+    "_class_name": "UNet2DConditionModel",
+    "config": "unet/config.json",
+    "weights": "unet/diffusion_pytorch_model.safetensors"
+  },
+  "vae": {
+    "_class_name": "AutoencoderKL",
     "config": "vae/config.json",
     "weights": "vae/diffusion_pytorch_model.safetensors"
   },
-  "unet": {
-    "config": "config.json",
-    "weights": "diffusion_pytorch_model.safetensors"
-  },
   "text_encoder": {
+    "_class_name": "CLIPTextModel",
     "config": "text_encoder/config.json",
     "weights": "text_encoder/pytorch_model.bin"
   },
+  "tokenizer": {
+    "_class_name": "CLIPTokenizer",
+    "config": "tokenizer/tokenizer_config.json",
+    "merges": "tokenizer/merges.txt",
+    "vocab": "tokenizer/vocab.json"
+  },
   "scheduler": {
+    "_class_name": "LMSDiscreteScheduler",
     "config": "scheduler/scheduler_config.json"
   }
 }