Update model_index.json

model_index.json

@@ -1,32 +1,7 @@
 {
-  "_class_name": "StableDiffusionPipeline",
-  "custom_pipeline": "stable_diffusion_text2img",
-  "feature_extractor": {
-    "_class_name": "CLIPFeatureExtractor",
-    "_diffusers_version": "0.21.4",
-    "do_normalize": true,
-    "do_resize": true,
-    "feature_extractor_type": "CLIPFeatureExtractor",
-    "image_mean": [0.48145466, 0.4578275, 0.40821073],
-    "image_std": [0.26862954, 0.26130258, 0.27577711],
-    "resample": 3,
-    "size": 224
-  },
-  "requires_safety_checker": false,
-  "safety_checker": null,
+  "_class_name": "StableDiffusionImg2ImgPipeline",
   "scheduler": {
-    "_class_name": "DDIMScheduler",
-    "_diffusers_version": "0.21.4",
-    "beta_end": 0.012,
-    "beta_schedule": "scaled_linear",
-    "beta_start": 0.00085,
-    "clip_sample": false,
-    "num_train_timesteps": 1000,
-    "prediction_type": "epsilon",
-    "set_alpha_to_one": false,
-    "skip_prk_steps": true,
-    "steps_offset": 1,
-    "trained_betas": null
+    "_class_name": "DDIMScheduler"
   },
   "text_encoder": {
     "_class_name": "CLIPTextModel"
@@ -34,11 +9,12 @@
   "tokenizer": {
     "_class_name": "CLIPTokenizer"
   },
-  "torch_dtype": "float16",
   "unet": {
     "_class_name": "UNet2DConditionModel"
   },
   "vae": {
     "_class_name": "AutoencoderKL"
-  }
+  },
+  "requires_safety_checker": false,
+  "safety_checker": null
 }