somq commited on
Commit
607cb5c
·
1 Parent(s): 1fdb17a

Revert "chore: disable safety checker & bump versions"

Browse files

This reverts commit 1fdb17a069391cde04cf676f1c0a3ade4f2f3d67.

model_index.json CHANGED
@@ -1,12 +1,33 @@
1
  {
2
  "_class_name": "StableDiffusionPipeline",
3
- "_diffusers_version": "0.24.0",
4
- "feature_extractor": ["transformers", "CLIPFeatureExtractor"],
5
- "requires_safety_checker": false,
6
- "safety_checker": ["stable_diffusion", "StableDiffusionSafetyChecker"],
7
- "scheduler": ["diffusers", "DPMSolverMultistepScheduler"],
8
- "text_encoder": ["transformers", "CLIPTextModel"],
9
- "tokenizer": ["transformers", "CLIPTokenizer"],
10
- "unet": ["diffusers", "UNet2DConditionModel"],
11
- "vae": ["diffusers", "AutoencoderKL"]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
12
  }
 
1
  {
2
  "_class_name": "StableDiffusionPipeline",
3
+ "_diffusers_version": "0.17.0.dev0",
4
+ "feature_extractor": [
5
+ "transformers",
6
+ "CLIPFeatureExtractor"
7
+ ],
8
+ "requires_safety_checker": true,
9
+ "safety_checker": [
10
+ "stable_diffusion",
11
+ "StableDiffusionSafetyChecker"
12
+ ],
13
+ "scheduler": [
14
+ "diffusers",
15
+ "DPMSolverMultistepScheduler"
16
+ ],
17
+ "text_encoder": [
18
+ "transformers",
19
+ "CLIPTextModel"
20
+ ],
21
+ "tokenizer": [
22
+ "transformers",
23
+ "CLIPTokenizer"
24
+ ],
25
+ "unet": [
26
+ "diffusers",
27
+ "UNet2DConditionModel"
28
+ ],
29
+ "vae": [
30
+ "diffusers",
31
+ "AutoencoderKL"
32
+ ]
33
  }
safety_checker/config.json CHANGED
@@ -1,7 +1,9 @@
1
  {
2
  "_commit_hash": "cb41f3a270d63d454d385fc2e4f571c487c253c5",
3
  "_name_or_path": "CompVis/stable-diffusion-safety-checker",
4
- "architectures": ["StableDiffusionSafetyChecker"],
 
 
5
  "initializer_factor": 1.0,
6
  "logit_scale_init_value": 2.6592,
7
  "model_type": "clip",
@@ -42,7 +44,7 @@
42
  "LABEL_0": 0,
43
  "LABEL_1": 1
44
  },
45
- "layer_norm_eps": 1e-5,
46
  "length_penalty": 1.0,
47
  "max_length": 20,
48
  "max_position_embeddings": 77,
@@ -78,7 +80,7 @@
78
  "top_p": 1.0,
79
  "torch_dtype": null,
80
  "torchscript": false,
81
- "transformers_version": "4.35.2",
82
  "typical_p": 1.0,
83
  "use_bfloat16": false,
84
  "vocab_size": 49408
@@ -122,7 +124,7 @@
122
  "LABEL_0": 0,
123
  "LABEL_1": 1
124
  },
125
- "layer_norm_eps": 1e-5,
126
  "length_penalty": 1.0,
127
  "max_length": 20,
128
  "min_length": 0,
@@ -159,7 +161,7 @@
159
  "top_p": 1.0,
160
  "torch_dtype": null,
161
  "torchscript": false,
162
- "transformers_version": "4.35.2",
163
  "typical_p": 1.0,
164
  "use_bfloat16": false
165
  }
 
1
  {
2
  "_commit_hash": "cb41f3a270d63d454d385fc2e4f571c487c253c5",
3
  "_name_or_path": "CompVis/stable-diffusion-safety-checker",
4
+ "architectures": [
5
+ "StableDiffusionSafetyChecker"
6
+ ],
7
  "initializer_factor": 1.0,
8
  "logit_scale_init_value": 2.6592,
9
  "model_type": "clip",
 
44
  "LABEL_0": 0,
45
  "LABEL_1": 1
46
  },
47
+ "layer_norm_eps": 1e-05,
48
  "length_penalty": 1.0,
49
  "max_length": 20,
50
  "max_position_embeddings": 77,
 
80
  "top_p": 1.0,
81
  "torch_dtype": null,
82
  "torchscript": false,
83
+ "transformers_version": "4.28.1",
84
  "typical_p": 1.0,
85
  "use_bfloat16": false,
86
  "vocab_size": 49408
 
124
  "LABEL_0": 0,
125
  "LABEL_1": 1
126
  },
127
+ "layer_norm_eps": 1e-05,
128
  "length_penalty": 1.0,
129
  "max_length": 20,
130
  "min_length": 0,
 
161
  "top_p": 1.0,
162
  "torch_dtype": null,
163
  "torchscript": false,
164
+ "transformers_version": "4.28.1",
165
  "typical_p": 1.0,
166
  "use_bfloat16": false
167
  }
scheduler/scheduler_config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_class_name": "DPMSolverMultistepScheduler",
3
- "_diffusers_version": "0.24.0",
4
  "algorithm_type": "dpmsolver++",
5
  "beta_end": 0.012,
6
  "beta_schedule": "scaled_linear",
 
1
  {
2
  "_class_name": "DPMSolverMultistepScheduler",
3
+ "_diffusers_version": "0.17.0.dev0",
4
  "algorithm_type": "dpmsolver++",
5
  "beta_end": 0.012,
6
  "beta_schedule": "scaled_linear",
text_encoder/config.json CHANGED
@@ -1,6 +1,8 @@
1
  {
2
  "_name_or_path": "openai/clip-vit-large-patch14",
3
- "architectures": ["CLIPTextModel"],
 
 
4
  "attention_dropout": 0.0,
5
  "bos_token_id": 0,
6
  "dropout": 0.0,
@@ -10,7 +12,7 @@
10
  "initializer_factor": 1.0,
11
  "initializer_range": 0.02,
12
  "intermediate_size": 3072,
13
- "layer_norm_eps": 1e-5,
14
  "max_position_embeddings": 77,
15
  "model_type": "clip_text_model",
16
  "num_attention_heads": 12,
@@ -18,6 +20,6 @@
18
  "pad_token_id": 1,
19
  "projection_dim": 768,
20
  "torch_dtype": "float16",
21
- "transformers_version": "4.35.2",
22
  "vocab_size": 49408
23
  }
 
1
  {
2
  "_name_or_path": "openai/clip-vit-large-patch14",
3
+ "architectures": [
4
+ "CLIPTextModel"
5
+ ],
6
  "attention_dropout": 0.0,
7
  "bos_token_id": 0,
8
  "dropout": 0.0,
 
12
  "initializer_factor": 1.0,
13
  "initializer_range": 0.02,
14
  "intermediate_size": 3072,
15
+ "layer_norm_eps": 1e-05,
16
  "max_position_embeddings": 77,
17
  "model_type": "clip_text_model",
18
  "num_attention_heads": 12,
 
20
  "pad_token_id": 1,
21
  "projection_dim": 768,
22
  "torch_dtype": "float16",
23
+ "transformers_version": "4.28.1",
24
  "vocab_size": 49408
25
  }
unet/config.json CHANGED
@@ -1,11 +1,16 @@
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
- "_diffusers_version": "0.24.0",
4
  "act_fn": "silu",
5
  "addition_embed_type": null,
6
  "addition_embed_type_num_heads": 64,
7
  "attention_head_dim": 8,
8
- "block_out_channels": [320, 640, 1280, 1280],
 
 
 
 
 
9
  "center_input_sample": false,
10
  "class_embed_type": null,
11
  "class_embeddings_concat": false,
@@ -29,7 +34,7 @@
29
  "mid_block_only_cross_attention": null,
30
  "mid_block_scale_factor": 1,
31
  "mid_block_type": "UNetMidBlock2DCrossAttn",
32
- "norm_eps": 1e-5,
33
  "norm_num_groups": 32,
34
  "num_class_embeds": null,
35
  "only_cross_attention": false,
 
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.17.0.dev0",
4
  "act_fn": "silu",
5
  "addition_embed_type": null,
6
  "addition_embed_type_num_heads": 64,
7
  "attention_head_dim": 8,
8
+ "block_out_channels": [
9
+ 320,
10
+ 640,
11
+ 1280,
12
+ 1280
13
+ ],
14
  "center_input_sample": false,
15
  "class_embed_type": null,
16
  "class_embeddings_concat": false,
 
34
  "mid_block_only_cross_attention": null,
35
  "mid_block_scale_factor": 1,
36
  "mid_block_type": "UNetMidBlock2DCrossAttn",
37
+ "norm_eps": 1e-05,
38
  "norm_num_groups": 32,
39
  "num_class_embeds": null,
40
  "only_cross_attention": false,
vae/config.json CHANGED
@@ -1,8 +1,13 @@
1
  {
2
  "_class_name": "AutoencoderKL",
3
- "_diffusers_version": "0.24.0",
4
  "act_fn": "silu",
5
- "block_out_channels": [128, 256, 512, 512],
 
 
 
 
 
6
  "down_block_types": [
7
  "DownEncoderBlock2D",
8
  "DownEncoderBlock2D",
 
1
  {
2
  "_class_name": "AutoencoderKL",
3
+ "_diffusers_version": "0.17.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",