etiennefd
/

tonalamatl_diffusion

StableDiffusionPipeline

Model card Files Files and versions

etiennefd commited on Nov 6, 2022

Commit

557501d

·

1 Parent(s): 060e165

Update model_index.json

Files changed (1) hide show

model_index.json +14 -6

model_index.json CHANGED Viewed

@@ -1,13 +1,21 @@
 {
-  "_class_name": "LDMTextToImagePipeline",
   "_diffusers_version": "0.3.0",
-  "bert": [
     "transformers",
-    "CLIPTextModel"
   ],
   "scheduler": [
     "diffusers",
-    "DDIMScheduler"
   ],
   "tokenizer": [
     "transformers",
@@ -17,8 +25,8 @@
     "diffusers",
     "UNet2DConditionModel"
   ],
-  "vqvae": [
     "diffusers",
     "AutoencoderKL"
   ]
-}

 {
+  "_class_name": "StableDiffusionPipeline",
   "_diffusers_version": "0.3.0",
+  "feature_extractor": [
     "transformers",
+    "CLIPFeatureExtractor"
+  ],
+  "safety_checker": [
+    "stable_diffusion",
+    "StableDiffusionSafetyChecker"
   ],
   "scheduler": [
     "diffusers",
+    "PNDMScheduler"
+  ],
+  "text_encoder": [
+    "transformers",
+    "CLIPTextModel"
   ],
   "tokenizer": [
     "transformers",
     "diffusers",
     "UNet2DConditionModel"
   ],
+  "vae": [
     "diffusers",
     "AutoencoderKL"
   ]
+}