Create model_index.json

model_index.json

@@ -0,0 +1,44 @@
+{
+  "_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,
+  "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
+  },
+  "text_encoder": {
+    "_class_name": "CLIPTextModel"
+  },
+  "tokenizer": {
+    "_class_name": "CLIPTokenizer"
+  },
+  "torch_dtype": "float16",
+  "unet": {
+    "_class_name": "UNet2DConditionModel"
+  },
+  "vae": {
+    "_class_name": "AutoencoderKL"
+  }
+}