etiennefd commited on
Commit
557501d
1 Parent(s): 060e165

Update model_index.json

Browse files
Files changed (1) hide show
  1. model_index.json +14 -6
model_index.json CHANGED
@@ -1,13 +1,21 @@
1
  {
2
- "_class_name": "LDMTextToImagePipeline",
3
  "_diffusers_version": "0.3.0",
4
- "bert": [
5
  "transformers",
6
- "CLIPTextModel"
 
 
 
 
7
  ],
8
  "scheduler": [
9
  "diffusers",
10
- "DDIMScheduler"
 
 
 
 
11
  ],
12
  "tokenizer": [
13
  "transformers",
@@ -17,8 +25,8 @@
17
  "diffusers",
18
  "UNet2DConditionModel"
19
  ],
20
- "vqvae": [
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
+ }