Adding `diffusers` weights of this model

#2
by Yntec - opened
Files changed (2) hide show
  1. model_index.json +3 -3
  2. vae/config.json +30 -0
model_index.json CHANGED
@@ -5,10 +5,10 @@
5
  "transformers",
6
  "CLIPFeatureExtractor"
7
  ],
8
- "requires_safety_checker": null,
9
  "safety_checker": [
10
- null,
11
- null
12
  ],
13
  "scheduler": [
14
  "diffusers",
 
5
  "transformers",
6
  "CLIPFeatureExtractor"
7
  ],
8
+ "requires_safety_checker": true,
9
  "safety_checker": [
10
+ "stable_diffusion",
11
+ "StableDiffusionSafetyChecker"
12
  ],
13
  "scheduler": [
14
  "diffusers",
vae/config.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "AutoencoderKL",
3
+ "_diffusers_version": "0.18.0.dev0",
4
+ "act_fn": "silu",
5
+ "block_out_channels": [
6
+ 128,
7
+ 256,
8
+ 512,
9
+ 512
10
+ ],
11
+ "down_block_types": [
12
+ "DownEncoderBlock2D",
13
+ "DownEncoderBlock2D",
14
+ "DownEncoderBlock2D",
15
+ "DownEncoderBlock2D"
16
+ ],
17
+ "in_channels": 3,
18
+ "latent_channels": 4,
19
+ "layers_per_block": 2,
20
+ "norm_num_groups": 32,
21
+ "out_channels": 3,
22
+ "sample_size": 768,
23
+ "scaling_factor": 0.18215,
24
+ "up_block_types": [
25
+ "UpDecoderBlock2D",
26
+ "UpDecoderBlock2D",
27
+ "UpDecoderBlock2D",
28
+ "UpDecoderBlock2D"
29
+ ]
30
+ }