Add feature extractor and safety checker
Browse files- model_index.json +5 -5
model_index.json
CHANGED
@@ -2,13 +2,13 @@
|
|
2 |
"_class_name": "StableDiffusionPipeline",
|
3 |
"_diffusers_version": "0.11.1",
|
4 |
"feature_extractor": [
|
5 |
-
|
6 |
-
|
7 |
],
|
8 |
-
"requires_safety_checker":
|
9 |
"safety_checker": [
|
10 |
-
|
11 |
-
|
12 |
],
|
13 |
"scheduler": [
|
14 |
"diffusers",
|
|
|
2 |
"_class_name": "StableDiffusionPipeline",
|
3 |
"_diffusers_version": "0.11.1",
|
4 |
"feature_extractor": [
|
5 |
+
"transformers",
|
6 |
+
"CLIPImageProcessor"
|
7 |
],
|
8 |
+
"requires_safety_checker": true,
|
9 |
"safety_checker": [
|
10 |
+
"stable_diffusion",
|
11 |
+
"StableDiffusionSafetyChecker"
|
12 |
],
|
13 |
"scheduler": [
|
14 |
"diffusers",
|