chamuditha4 commited on
Commit
ccfc656
·
verified ·
1 Parent(s): 2c8d7d9

Upload model

Browse files
feature_extractor/preprocessor_config.json CHANGED
@@ -8,7 +8,6 @@
8
  "do_normalize": true,
9
  "do_rescale": true,
10
  "do_resize": true,
11
- "feature_extractor_type": "CLIPFeatureExtractor",
12
  "image_mean": [
13
  0.48145466,
14
  0.4578275,
 
8
  "do_normalize": true,
9
  "do_rescale": true,
10
  "do_resize": true,
 
11
  "image_mean": [
12
  0.48145466,
13
  0.4578275,
model_index.json CHANGED
@@ -1,10 +1,15 @@
1
  {
2
  "_class_name": "StableDiffusionPipeline",
3
- "_diffusers_version": "0.20.0",
 
4
  "feature_extractor": [
5
  "transformers",
6
  "CLIPFeatureExtractor"
7
  ],
 
 
 
 
8
  "requires_safety_checker": true,
9
  "safety_checker": [
10
  "stable_diffusion",
 
1
  {
2
  "_class_name": "StableDiffusionPipeline",
3
+ "_diffusers_version": "0.30.0",
4
+ "_name_or_path": "/content/output_model",
5
  "feature_extractor": [
6
  "transformers",
7
  "CLIPFeatureExtractor"
8
  ],
9
+ "image_encoder": [
10
+ null,
11
+ null
12
+ ],
13
  "requires_safety_checker": true,
14
  "safety_checker": [
15
  "stable_diffusion",
safety_checker/config.json CHANGED
@@ -1,6 +1,5 @@
1
  {
2
- "_commit_hash": "cb41f3a270d63d454d385fc2e4f571c487c253c5",
3
- "_name_or_path": "CompVis/stable-diffusion-safety-checker",
4
  "architectures": [
5
  "StableDiffusionSafetyChecker"
6
  ],
@@ -9,160 +8,21 @@
9
  "model_type": "clip",
10
  "projection_dim": 768,
11
  "text_config": {
12
- "_name_or_path": "",
13
- "add_cross_attention": false,
14
- "architectures": null,
15
- "attention_dropout": 0.0,
16
- "bad_words_ids": null,
17
- "begin_suppress_tokens": null,
18
- "bos_token_id": 49406,
19
- "chunk_size_feed_forward": 0,
20
- "cross_attention_hidden_size": null,
21
- "decoder_start_token_id": null,
22
- "diversity_penalty": 0.0,
23
- "do_sample": false,
24
  "dropout": 0.0,
25
- "early_stopping": false,
26
- "encoder_no_repeat_ngram_size": 0,
27
- "eos_token_id": 49407,
28
- "exponential_decay_length_penalty": null,
29
- "finetuning_task": null,
30
- "forced_bos_token_id": null,
31
- "forced_eos_token_id": null,
32
- "hidden_act": "quick_gelu",
33
  "hidden_size": 768,
34
- "id2label": {
35
- "0": "LABEL_0",
36
- "1": "LABEL_1"
37
- },
38
- "initializer_factor": 1.0,
39
- "initializer_range": 0.02,
40
  "intermediate_size": 3072,
41
- "is_decoder": false,
42
- "is_encoder_decoder": false,
43
- "label2id": {
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,
51
- "min_length": 0,
52
  "model_type": "clip_text_model",
53
- "no_repeat_ngram_size": 0,
54
- "num_attention_heads": 12,
55
- "num_beam_groups": 1,
56
- "num_beams": 1,
57
- "num_hidden_layers": 12,
58
- "num_return_sequences": 1,
59
- "output_attentions": false,
60
- "output_hidden_states": false,
61
- "output_scores": false,
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,
69
- "return_dict": true,
70
- "return_dict_in_generate": false,
71
- "sep_token_id": null,
72
- "suppress_tokens": null,
73
- "task_specific_params": null,
74
- "temperature": 1.0,
75
- "tf_legacy_loss": false,
76
- "tie_encoder_decoder": false,
77
- "tie_word_embeddings": true,
78
- "tokenizer_class": null,
79
- "top_k": 50,
80
- "top_p": 1.0,
81
- "torch_dtype": null,
82
- "torchscript": false,
83
- "transformers_version": "4.31.0",
84
- "typical_p": 1.0,
85
- "use_bfloat16": false,
86
- "vocab_size": 49408
87
  },
88
- "torch_dtype": "float32",
89
- "transformers_version": null,
90
  "vision_config": {
91
- "_name_or_path": "",
92
- "add_cross_attention": false,
93
- "architectures": null,
94
- "attention_dropout": 0.0,
95
- "bad_words_ids": null,
96
- "begin_suppress_tokens": null,
97
- "bos_token_id": null,
98
- "chunk_size_feed_forward": 0,
99
- "cross_attention_hidden_size": null,
100
- "decoder_start_token_id": null,
101
- "diversity_penalty": 0.0,
102
- "do_sample": false,
103
  "dropout": 0.0,
104
- "early_stopping": false,
105
- "encoder_no_repeat_ngram_size": 0,
106
- "eos_token_id": null,
107
- "exponential_decay_length_penalty": null,
108
- "finetuning_task": null,
109
- "forced_bos_token_id": null,
110
- "forced_eos_token_id": null,
111
- "hidden_act": "quick_gelu",
112
  "hidden_size": 1024,
113
- "id2label": {
114
- "0": "LABEL_0",
115
- "1": "LABEL_1"
116
- },
117
- "image_size": 224,
118
- "initializer_factor": 1.0,
119
- "initializer_range": 0.02,
120
  "intermediate_size": 4096,
121
- "is_decoder": false,
122
- "is_encoder_decoder": false,
123
- "label2id": {
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,
131
  "model_type": "clip_vision_model",
132
- "no_repeat_ngram_size": 0,
133
  "num_attention_heads": 16,
134
- "num_beam_groups": 1,
135
- "num_beams": 1,
136
- "num_channels": 3,
137
  "num_hidden_layers": 24,
138
- "num_return_sequences": 1,
139
- "output_attentions": false,
140
- "output_hidden_states": false,
141
- "output_scores": false,
142
- "pad_token_id": null,
143
- "patch_size": 14,
144
- "prefix": null,
145
- "problem_type": null,
146
- "projection_dim": 512,
147
- "pruned_heads": {},
148
- "remove_invalid_values": false,
149
- "repetition_penalty": 1.0,
150
- "return_dict": true,
151
- "return_dict_in_generate": false,
152
- "sep_token_id": null,
153
- "suppress_tokens": null,
154
- "task_specific_params": null,
155
- "temperature": 1.0,
156
- "tf_legacy_loss": false,
157
- "tie_encoder_decoder": false,
158
- "tie_word_embeddings": true,
159
- "tokenizer_class": null,
160
- "top_k": 50,
161
- "top_p": 1.0,
162
- "torch_dtype": null,
163
- "torchscript": false,
164
- "transformers_version": "4.31.0",
165
- "typical_p": 1.0,
166
- "use_bfloat16": false
167
  }
168
  }
 
1
  {
2
+ "_name_or_path": "/content/output_model/safety_checker",
 
3
  "architectures": [
4
  "StableDiffusionSafetyChecker"
5
  ],
 
8
  "model_type": "clip",
9
  "projection_dim": 768,
10
  "text_config": {
 
 
 
 
 
 
 
 
 
 
 
 
11
  "dropout": 0.0,
 
 
 
 
 
 
 
 
12
  "hidden_size": 768,
 
 
 
 
 
 
13
  "intermediate_size": 3072,
 
 
 
 
 
 
 
 
 
 
 
14
  "model_type": "clip_text_model",
15
+ "num_attention_heads": 12
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
16
  },
17
+ "torch_dtype": "float16",
18
+ "transformers_version": "4.42.4",
19
  "vision_config": {
 
 
 
 
 
 
 
 
 
 
 
 
20
  "dropout": 0.0,
 
 
 
 
 
 
 
 
21
  "hidden_size": 1024,
 
 
 
 
 
 
 
22
  "intermediate_size": 4096,
 
 
 
 
 
 
 
 
 
 
23
  "model_type": "clip_vision_model",
 
24
  "num_attention_heads": 16,
 
 
 
25
  "num_hidden_layers": 24,
26
+ "patch_size": 14
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
27
  }
28
  }
safety_checker/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:57ecdfa243b170f9b4cb3eefaf0f64552ef78fc0bf0eb1c5b9675308447184f6
3
+ size 608016280
scheduler/scheduler_config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_class_name": "PNDMScheduler",
3
- "_diffusers_version": "0.20.0",
4
  "beta_end": 0.012,
5
  "beta_schedule": "scaled_linear",
6
  "beta_start": 0.00085,
 
1
  {
2
  "_class_name": "PNDMScheduler",
3
+ "_diffusers_version": "0.30.0",
4
  "beta_end": 0.012,
5
  "beta_schedule": "scaled_linear",
6
  "beta_start": 0.00085,
text_encoder/config.json CHANGED
@@ -1,4 +1,5 @@
1
  {
 
2
  "architectures": [
3
  "CLIPTextModel"
4
  ],
@@ -18,7 +19,7 @@
18
  "num_hidden_layers": 12,
19
  "pad_token_id": 1,
20
  "projection_dim": 768,
21
- "torch_dtype": "float32",
22
- "transformers_version": "4.31.0",
23
  "vocab_size": 49408
24
  }
 
1
  {
2
+ "_name_or_path": "/content/output_model/text_encoder",
3
  "architectures": [
4
  "CLIPTextModel"
5
  ],
 
19
  "num_hidden_layers": 12,
20
  "pad_token_id": 1,
21
  "projection_dim": 768,
22
+ "torch_dtype": "float16",
23
+ "transformers_version": "4.42.4",
24
  "vocab_size": 49408
25
  }
text_encoder/model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b4f27c4bde308a8aa337ce755e2af288757d86309d74b29b0bd1a3f5d6092864
3
- size 492265168
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f79bec2680258e5782376bce862507670c142330b21f3a0426522157310c3383
3
+ size 246144152
tokenizer/special_tokens_map.json CHANGED
@@ -9,15 +9,21 @@
9
  "eos_token": {
10
  "content": "<|endoftext|>",
11
  "lstrip": false,
12
- "normalized": true,
 
 
 
 
 
 
 
13
  "rstrip": false,
14
  "single_word": false
15
  },
16
- "pad_token": "<|endoftext|>",
17
  "unk_token": {
18
  "content": "<|endoftext|>",
19
  "lstrip": false,
20
- "normalized": true,
21
  "rstrip": false,
22
  "single_word": false
23
  }
 
9
  "eos_token": {
10
  "content": "<|endoftext|>",
11
  "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": {
17
+ "content": "<|endoftext|>",
18
+ "lstrip": false,
19
+ "normalized": false,
20
  "rstrip": false,
21
  "single_word": false
22
  },
 
23
  "unk_token": {
24
  "content": "<|endoftext|>",
25
  "lstrip": false,
26
+ "normalized": false,
27
  "rstrip": false,
28
  "single_word": false
29
  }
tokenizer/tokenizer_config.json CHANGED
@@ -1,33 +1,30 @@
1
  {
2
  "add_prefix_space": false,
3
- "bos_token": {
4
- "__type": "AddedToken",
5
- "content": "<|startoftext|>",
6
- "lstrip": false,
7
- "normalized": true,
8
- "rstrip": false,
9
- "single_word": false
 
 
 
 
 
 
 
 
 
 
10
  },
 
11
  "clean_up_tokenization_spaces": true,
12
  "do_lower_case": true,
13
- "eos_token": {
14
- "__type": "AddedToken",
15
- "content": "<|endoftext|>",
16
- "lstrip": false,
17
- "normalized": true,
18
- "rstrip": false,
19
- "single_word": false
20
- },
21
  "errors": "replace",
22
  "model_max_length": 77,
23
  "pad_token": "<|endoftext|>",
24
  "tokenizer_class": "CLIPTokenizer",
25
- "unk_token": {
26
- "__type": "AddedToken",
27
- "content": "<|endoftext|>",
28
- "lstrip": false,
29
- "normalized": true,
30
- "rstrip": false,
31
- "single_word": false
32
- }
33
  }
 
1
  {
2
  "add_prefix_space": false,
3
+ "added_tokens_decoder": {
4
+ "49406": {
5
+ "content": "<|startoftext|>",
6
+ "lstrip": false,
7
+ "normalized": true,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
+ "49407": {
13
+ "content": "<|endoftext|>",
14
+ "lstrip": false,
15
+ "normalized": false,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ }
20
  },
21
+ "bos_token": "<|startoftext|>",
22
  "clean_up_tokenization_spaces": true,
23
  "do_lower_case": true,
24
+ "eos_token": "<|endoftext|>",
 
 
 
 
 
 
 
25
  "errors": "replace",
26
  "model_max_length": 77,
27
  "pad_token": "<|endoftext|>",
28
  "tokenizer_class": "CLIPTokenizer",
29
+ "unk_token": "<|endoftext|>"
 
 
 
 
 
 
 
30
  }
unet/config.json CHANGED
@@ -1,6 +1,7 @@
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
- "_diffusers_version": "0.20.0",
 
4
  "act_fn": "silu",
5
  "addition_embed_type": null,
6
  "addition_embed_type_num_heads": 64,
@@ -27,6 +28,7 @@
27
  "DownBlock2D"
28
  ],
29
  "downsample_padding": 1,
 
30
  "dual_cross_attention": false,
31
  "encoder_hid_dim": null,
32
  "encoder_hid_dim_type": null,
@@ -47,6 +49,7 @@
47
  "resnet_out_scale_factor": 1.0,
48
  "resnet_skip_time_act": false,
49
  "resnet_time_scale_shift": "default",
 
50
  "sample_size": 64,
51
  "time_cond_proj_dim": null,
52
  "time_embedding_act_fn": null,
 
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.30.0",
4
+ "_name_or_path": "/content/output_model/unet",
5
  "act_fn": "silu",
6
  "addition_embed_type": null,
7
  "addition_embed_type_num_heads": 64,
 
28
  "DownBlock2D"
29
  ],
30
  "downsample_padding": 1,
31
+ "dropout": 0.0,
32
  "dual_cross_attention": false,
33
  "encoder_hid_dim": null,
34
  "encoder_hid_dim_type": null,
 
49
  "resnet_out_scale_factor": 1.0,
50
  "resnet_skip_time_act": false,
51
  "resnet_time_scale_shift": "default",
52
+ "reverse_transformer_layers_per_block": null,
53
  "sample_size": 64,
54
  "time_cond_proj_dim": null,
55
  "time_embedding_act_fn": null,
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4e0868e8fcae7d4ea8f8cdd3051704b3b47d741dc8e8629552d1a07f6efb8e32
3
- size 3438167536
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aafe70e42d399738ac2c49e9447ee088c6b87fbc23e605452756c33ce9ffc0eb
3
+ size 1719125304
vae/config.json CHANGED
@@ -1,6 +1,7 @@
1
  {
2
  "_class_name": "AutoencoderKL",
3
- "_diffusers_version": "0.20.0",
 
4
  "act_fn": "silu",
5
  "block_out_channels": [
6
  128,
@@ -17,15 +18,21 @@
17
  "force_upcast": true,
18
  "in_channels": 3,
19
  "latent_channels": 4,
 
 
20
  "layers_per_block": 2,
 
21
  "norm_num_groups": 32,
22
  "out_channels": 3,
23
  "sample_size": 512,
24
  "scaling_factor": 0.18215,
 
25
  "up_block_types": [
26
  "UpDecoderBlock2D",
27
  "UpDecoderBlock2D",
28
  "UpDecoderBlock2D",
29
  "UpDecoderBlock2D"
30
- ]
 
 
31
  }
 
1
  {
2
  "_class_name": "AutoencoderKL",
3
+ "_diffusers_version": "0.30.0",
4
+ "_name_or_path": "/content/output_model/vae",
5
  "act_fn": "silu",
6
  "block_out_channels": [
7
  128,
 
18
  "force_upcast": true,
19
  "in_channels": 3,
20
  "latent_channels": 4,
21
+ "latents_mean": null,
22
+ "latents_std": null,
23
  "layers_per_block": 2,
24
+ "mid_block_add_attention": true,
25
  "norm_num_groups": 32,
26
  "out_channels": 3,
27
  "sample_size": 512,
28
  "scaling_factor": 0.18215,
29
+ "shift_factor": null,
30
  "up_block_types": [
31
  "UpDecoderBlock2D",
32
  "UpDecoderBlock2D",
33
  "UpDecoderBlock2D",
34
  "UpDecoderBlock2D"
35
+ ],
36
+ "use_post_quant_conv": true,
37
+ "use_quant_conv": true
38
  }
vae/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:86a6170968f284b6c0c111e2611c806a04102e2a48790ebd1d490ca78040d2bb
3
- size 334643268
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3e4c08995484ee61270175e9e7a072b66a6e4eeb5f0c266667fe1f45b90daf9a
3
+ size 167335342