Update model_index.json

model_index.json

@@ -1,13 +1,28 @@
 {
-  "pipeline_class": "StableDiffusionPipeline",
+  "_class_name": "StableDiffusionPipeline",
   "model_type": "ImageGeneration",
   "base_model": "UNet",
   "scheduler": "DDIMScheduler",
   "components": {
-    "unet": "unet/unet.bin",
-    "vae": "vae/vae.bin",
-    "scheduler": "scheduler/scheduler.json",
-    "text_encoder": "text_encoder/text_encoder.bin",
-    "tokenizer": "tokenizer/tokenizer.json"
+    "unet": {
+      "_class_name": "UNet2DConditionModel",
+      "file": "unet/unet.bin"
+    },
+    "vae": {
+      "_class_name": "AutoencoderKL",
+      "file": "vae/vae.bin"
+    },
+    "scheduler": {
+      "_class_name": "DDIMScheduler",
+      "file": "scheduler/scheduler.json"
+    },
+    "text_encoder": {
+      "_class_name": "CLIPTextModel",
+      "file": "text_encoder/text_encoder.bin"
+    },
+    "tokenizer": {
+      "_class_name": "CLIPTokenizer",
+      "file": "tokenizer/tokenizer.json"
+    }
   }
 }