Update model_index.json
Browse files- model_index.json +14 -6
model_index.json
CHANGED
@@ -1,13 +1,21 @@
|
|
1 |
{
|
2 |
-
"_class_name": "
|
3 |
"_diffusers_version": "0.3.0",
|
4 |
-
"
|
5 |
"transformers",
|
6 |
-
"
|
|
|
|
|
|
|
|
|
7 |
],
|
8 |
"scheduler": [
|
9 |
"diffusers",
|
10 |
-
"
|
|
|
|
|
|
|
|
|
11 |
],
|
12 |
"tokenizer": [
|
13 |
"transformers",
|
@@ -17,8 +25,8 @@
|
|
17 |
"diffusers",
|
18 |
"UNet2DConditionModel"
|
19 |
],
|
20 |
-
"
|
21 |
"diffusers",
|
22 |
"AutoencoderKL"
|
23 |
]
|
24 |
-
}
|
|
|
1 |
{
|
2 |
+
"_class_name": "StableDiffusionPipeline",
|
3 |
"_diffusers_version": "0.3.0",
|
4 |
+
"feature_extractor": [
|
5 |
"transformers",
|
6 |
+
"CLIPFeatureExtractor"
|
7 |
+
],
|
8 |
+
"safety_checker": [
|
9 |
+
"stable_diffusion",
|
10 |
+
"StableDiffusionSafetyChecker"
|
11 |
],
|
12 |
"scheduler": [
|
13 |
"diffusers",
|
14 |
+
"PNDMScheduler"
|
15 |
+
],
|
16 |
+
"text_encoder": [
|
17 |
+
"transformers",
|
18 |
+
"CLIPTextModel"
|
19 |
],
|
20 |
"tokenizer": [
|
21 |
"transformers",
|
|
|
25 |
"diffusers",
|
26 |
"UNet2DConditionModel"
|
27 |
],
|
28 |
+
"vae": [
|
29 |
"diffusers",
|
30 |
"AutoencoderKL"
|
31 |
]
|
32 |
+
}
|