sharpbai
/

stable-diffusion-v1-5-onnx-cuda-fp32

Model card Files Files and versions

sharpbai commited on Jul 14, 2023

Commit

ae5e637

·

1 Parent(s): c6c3b5f

Update model_index.json

Files changed (1) hide show

model_index.json +12 -8

model_index.json CHANGED Viewed

@@ -7,16 +7,16 @@
   ],
   "requires_safety_checker": true,
   "safety_checker": [
-    "stable_diffusion",
-    "StableDiffusionSafetyChecker"
   ],
   "scheduler": [
     "diffusers",
     "PNDMScheduler"
   ],
   "text_encoder": [
-    "transformers",
-    "CLIPTextModel"
   ],
   "tokenizer": [
     "transformers",
@@ -24,10 +24,14 @@
   ],
   "unet": [
     "diffusers",
-    "UNet2DConditionModel"
   ],
-  "vae": [
     "diffusers",
-    "AutoencoderKL"
   ]
-}

   ],
   "requires_safety_checker": true,
   "safety_checker": [
+    "diffusers",
+    "OnnxRuntimeModel"
   ],
   "scheduler": [
     "diffusers",
     "PNDMScheduler"
   ],
   "text_encoder": [
+    "diffusers",
+    "OnnxRuntimeModel"
   ],
   "tokenizer": [
     "transformers",
   ],
   "unet": [
     "diffusers",
+    "OnnxRuntimeModel"
+  ],
+  "vae_decoder": [
+    "diffusers",
+    "OnnxRuntimeModel"
   ],
+  "vae_encoder": [
     "diffusers",
+    "OnnxRuntimeModel"
   ]
+}