Conflictx Alex Poulos commited on
Commit
de8bbc7
1 Parent(s): 05515b8

Add Diffusers weights (#5)

Browse files

- Add Diffusers weights (2e36a6dc7bcde2158678496699630a2e6dd5d132)


Co-authored-by: Alex Poulos <[email protected]>

feature_extractor/preprocessor_config.json CHANGED
@@ -1,8 +1,12 @@
1
  {
2
- "crop_size": 224,
 
 
 
3
  "do_center_crop": true,
4
  "do_convert_rgb": true,
5
  "do_normalize": true,
 
6
  "do_resize": true,
7
  "feature_extractor_type": "CLIPFeatureExtractor",
8
  "image_mean": [
@@ -10,11 +14,15 @@
10
  0.4578275,
11
  0.40821073
12
  ],
 
13
  "image_std": [
14
  0.26862954,
15
  0.26130258,
16
  0.27577711
17
  ],
18
  "resample": 3,
19
- "size": 224
 
 
 
20
  }
 
1
  {
2
+ "crop_size": {
3
+ "height": 224,
4
+ "width": 224
5
+ },
6
  "do_center_crop": true,
7
  "do_convert_rgb": true,
8
  "do_normalize": true,
9
+ "do_rescale": true,
10
  "do_resize": true,
11
  "feature_extractor_type": "CLIPFeatureExtractor",
12
  "image_mean": [
 
14
  0.4578275,
15
  0.40821073
16
  ],
17
+ "image_processor_type": "CLIPImageProcessor",
18
  "image_std": [
19
  0.26862954,
20
  0.26130258,
21
  0.27577711
22
  ],
23
  "resample": 3,
24
+ "rescale_factor": 0.00392156862745098,
25
+ "size": {
26
+ "shortest_edge": 224
27
+ }
28
  }
model_index.json CHANGED
@@ -1,10 +1,11 @@
1
  {
2
  "_class_name": "StableDiffusionPipeline",
3
- "_diffusers_version": "0.8.0.dev0",
4
  "feature_extractor": [
5
  "transformers",
6
- "CLIPFeatureExtractor"
7
  ],
 
8
  "safety_checker": [
9
  "stable_diffusion",
10
  "StableDiffusionSafetyChecker"
 
1
  {
2
  "_class_name": "StableDiffusionPipeline",
3
+ "_diffusers_version": "0.10.0.dev0",
4
  "feature_extractor": [
5
  "transformers",
6
+ "CLIPImageProcessor"
7
  ],
8
+ "requires_safety_checker": true,
9
  "safety_checker": [
10
  "stable_diffusion",
11
  "StableDiffusionSafetyChecker"
safety_checker/config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_commit_hash": "4bb648a606ef040e7685bde262611766a5fdd67b",
3
  "_name_or_path": "CompVis/stable-diffusion-safety-checker",
4
  "architectures": [
5
  "StableDiffusionSafetyChecker"
@@ -62,6 +62,7 @@
62
  "pad_token_id": 1,
63
  "prefix": null,
64
  "problem_type": null,
 
65
  "pruned_heads": {},
66
  "remove_invalid_values": false,
67
  "repetition_penalty": 1.0,
@@ -79,7 +80,7 @@
79
  "top_p": 1.0,
80
  "torch_dtype": null,
81
  "torchscript": false,
82
- "transformers_version": "4.24.0",
83
  "typical_p": 1.0,
84
  "use_bfloat16": false,
85
  "vocab_size": 49408
@@ -148,6 +149,7 @@
148
  "patch_size": 14,
149
  "prefix": null,
150
  "problem_type": null,
 
151
  "pruned_heads": {},
152
  "remove_invalid_values": false,
153
  "repetition_penalty": 1.0,
@@ -165,7 +167,7 @@
165
  "top_p": 1.0,
166
  "torch_dtype": null,
167
  "torchscript": false,
168
- "transformers_version": "4.24.0",
169
  "typical_p": 1.0,
170
  "use_bfloat16": false
171
  },
 
1
  {
2
+ "_commit_hash": "cb41f3a270d63d454d385fc2e4f571c487c253c5",
3
  "_name_or_path": "CompVis/stable-diffusion-safety-checker",
4
  "architectures": [
5
  "StableDiffusionSafetyChecker"
 
62
  "pad_token_id": 1,
63
  "prefix": null,
64
  "problem_type": null,
65
+ "projection_dim": 512,
66
  "pruned_heads": {},
67
  "remove_invalid_values": false,
68
  "repetition_penalty": 1.0,
 
80
  "top_p": 1.0,
81
  "torch_dtype": null,
82
  "torchscript": false,
83
+ "transformers_version": "4.26.0.dev0",
84
  "typical_p": 1.0,
85
  "use_bfloat16": false,
86
  "vocab_size": 49408
 
149
  "patch_size": 14,
150
  "prefix": null,
151
  "problem_type": null,
152
+ "projection_dim": 512,
153
  "pruned_heads": {},
154
  "remove_invalid_values": false,
155
  "repetition_penalty": 1.0,
 
167
  "top_p": 1.0,
168
  "torch_dtype": null,
169
  "torchscript": false,
170
+ "transformers_version": "4.26.0.dev0",
171
  "typical_p": 1.0,
172
  "use_bfloat16": false
173
  },
scheduler/scheduler_config.json CHANGED
@@ -1,10 +1,12 @@
1
  {
2
  "_class_name": "PNDMScheduler",
3
- "_diffusers_version": "0.8.0.dev0",
4
  "beta_end": 0.012,
5
  "beta_schedule": "scaled_linear",
6
  "beta_start": 0.00085,
 
7
  "num_train_timesteps": 1000,
 
8
  "set_alpha_to_one": false,
9
  "skip_prk_steps": true,
10
  "steps_offset": 1,
 
1
  {
2
  "_class_name": "PNDMScheduler",
3
+ "_diffusers_version": "0.10.0.dev0",
4
  "beta_end": 0.012,
5
  "beta_schedule": "scaled_linear",
6
  "beta_start": 0.00085,
7
+ "clip_sample": false,
8
  "num_train_timesteps": 1000,
9
+ "prediction_type": "epsilon",
10
  "set_alpha_to_one": false,
11
  "skip_prk_steps": true,
12
  "steps_offset": 1,
text_encoder/config.json CHANGED
@@ -20,6 +20,6 @@
20
  "pad_token_id": 1,
21
  "projection_dim": 768,
22
  "torch_dtype": "float32",
23
- "transformers_version": "4.24.0",
24
  "vocab_size": 49408
25
  }
 
20
  "pad_token_id": 1,
21
  "projection_dim": 768,
22
  "torch_dtype": "float32",
23
+ "transformers_version": "4.26.0.dev0",
24
  "vocab_size": 49408
25
  }
unet/config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
- "_diffusers_version": "0.8.0.dev0",
4
  "act_fn": "silu",
5
  "attention_head_dim": 8,
6
  "block_out_channels": [
@@ -18,6 +18,7 @@
18
  "DownBlock2D"
19
  ],
20
  "downsample_padding": 1,
 
21
  "flip_sin_to_cos": true,
22
  "freq_shift": 0,
23
  "in_channels": 4,
@@ -25,6 +26,8 @@
25
  "mid_block_scale_factor": 1,
26
  "norm_eps": 1e-05,
27
  "norm_num_groups": 32,
 
 
28
  "out_channels": 4,
29
  "sample_size": 64,
30
  "up_block_types": [
@@ -32,5 +35,6 @@
32
  "CrossAttnUpBlock2D",
33
  "CrossAttnUpBlock2D",
34
  "CrossAttnUpBlock2D"
35
- ]
 
36
  }
 
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.10.0.dev0",
4
  "act_fn": "silu",
5
  "attention_head_dim": 8,
6
  "block_out_channels": [
 
18
  "DownBlock2D"
19
  ],
20
  "downsample_padding": 1,
21
+ "dual_cross_attention": false,
22
  "flip_sin_to_cos": true,
23
  "freq_shift": 0,
24
  "in_channels": 4,
 
26
  "mid_block_scale_factor": 1,
27
  "norm_eps": 1e-05,
28
  "norm_num_groups": 32,
29
+ "num_class_embeds": null,
30
+ "only_cross_attention": false,
31
  "out_channels": 4,
32
  "sample_size": 64,
33
  "up_block_types": [
 
35
  "CrossAttnUpBlock2D",
36
  "CrossAttnUpBlock2D",
37
  "CrossAttnUpBlock2D"
38
+ ],
39
+ "use_linear_projection": false
40
  }
vae/config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_class_name": "AutoencoderKL",
3
- "_diffusers_version": "0.8.0.dev0",
4
  "act_fn": "silu",
5
  "block_out_channels": [
6
  128,
@@ -19,7 +19,7 @@
19
  "layers_per_block": 2,
20
  "norm_num_groups": 32,
21
  "out_channels": 3,
22
- "sample_size": 256,
23
  "up_block_types": [
24
  "UpDecoderBlock2D",
25
  "UpDecoderBlock2D",
 
1
  {
2
  "_class_name": "AutoencoderKL",
3
+ "_diffusers_version": "0.10.0.dev0",
4
  "act_fn": "silu",
5
  "block_out_channels": [
6
  128,
 
19
  "layers_per_block": 2,
20
  "norm_num_groups": 32,
21
  "out_channels": 3,
22
+ "sample_size": 512,
23
  "up_block_types": [
24
  "UpDecoderBlock2D",
25
  "UpDecoderBlock2D",