safety_checker/config.json
4.7 KB · 178 lines · json Raw
1 {
2 "_commit_hash": "4bb648a606ef040e7685bde262611766a5fdd67b",
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",
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 "bos_token_id": 0,
18 "chunk_size_feed_forward": 0,
19 "cross_attention_hidden_size": null,
20 "decoder_start_token_id": null,
21 "diversity_penalty": 0.0,
22 "do_sample": false,
23 "dropout": 0.0,
24 "early_stopping": false,
25 "encoder_no_repeat_ngram_size": 0,
26 "eos_token_id": 2,
27 "exponential_decay_length_penalty": null,
28 "finetuning_task": null,
29 "forced_bos_token_id": null,
30 "forced_eos_token_id": null,
31 "hidden_act": "quick_gelu",
32 "hidden_size": 768,
33 "id2label": {
34 "0": "LABEL_0",
35 "1": "LABEL_1"
36 },
37 "initializer_factor": 1.0,
38 "initializer_range": 0.02,
39 "intermediate_size": 3072,
40 "is_decoder": false,
41 "is_encoder_decoder": false,
42 "label2id": {
43 "LABEL_0": 0,
44 "LABEL_1": 1
45 },
46 "layer_norm_eps": 1e-05,
47 "length_penalty": 1.0,
48 "max_length": 20,
49 "max_position_embeddings": 77,
50 "min_length": 0,
51 "model_type": "clip_text_model",
52 "no_repeat_ngram_size": 0,
53 "num_attention_heads": 12,
54 "num_beam_groups": 1,
55 "num_beams": 1,
56 "num_hidden_layers": 12,
57 "num_return_sequences": 1,
58 "output_attentions": false,
59 "output_hidden_states": false,
60 "output_scores": false,
61 "pad_token_id": 1,
62 "prefix": null,
63 "problem_type": null,
64 "projection_dim": 512,
65 "pruned_heads": {},
66 "remove_invalid_values": false,
67 "repetition_penalty": 1.0,
68 "return_dict": true,
69 "return_dict_in_generate": false,
70 "sep_token_id": null,
71 "task_specific_params": null,
72 "temperature": 1.0,
73 "tf_legacy_loss": false,
74 "tie_encoder_decoder": false,
75 "tie_word_embeddings": true,
76 "tokenizer_class": null,
77 "top_k": 50,
78 "top_p": 1.0,
79 "torch_dtype": null,
80 "torchscript": false,
81 "transformers_version": "4.22.0.dev0",
82 "typical_p": 1.0,
83 "use_bfloat16": false,
84 "vocab_size": 49408
85 },
86 "text_config_dict": {
87 "hidden_size": 768,
88 "intermediate_size": 3072,
89 "num_attention_heads": 12,
90 "num_hidden_layers": 12
91 },
92 "torch_dtype": "float32",
93 "transformers_version": null,
94 "vision_config": {
95 "_name_or_path": "",
96 "add_cross_attention": false,
97 "architectures": null,
98 "attention_dropout": 0.0,
99 "bad_words_ids": null,
100 "bos_token_id": null,
101 "chunk_size_feed_forward": 0,
102 "cross_attention_hidden_size": null,
103 "decoder_start_token_id": null,
104 "diversity_penalty": 0.0,
105 "do_sample": false,
106 "dropout": 0.0,
107 "early_stopping": false,
108 "encoder_no_repeat_ngram_size": 0,
109 "eos_token_id": null,
110 "exponential_decay_length_penalty": null,
111 "finetuning_task": null,
112 "forced_bos_token_id": null,
113 "forced_eos_token_id": null,
114 "hidden_act": "quick_gelu",
115 "hidden_size": 1024,
116 "id2label": {
117 "0": "LABEL_0",
118 "1": "LABEL_1"
119 },
120 "image_size": 224,
121 "initializer_factor": 1.0,
122 "initializer_range": 0.02,
123 "intermediate_size": 4096,
124 "is_decoder": false,
125 "is_encoder_decoder": false,
126 "label2id": {
127 "LABEL_0": 0,
128 "LABEL_1": 1
129 },
130 "layer_norm_eps": 1e-05,
131 "length_penalty": 1.0,
132 "max_length": 20,
133 "min_length": 0,
134 "model_type": "clip_vision_model",
135 "no_repeat_ngram_size": 0,
136 "num_attention_heads": 16,
137 "num_beam_groups": 1,
138 "num_beams": 1,
139 "num_channels": 3,
140 "num_hidden_layers": 24,
141 "num_return_sequences": 1,
142 "output_attentions": false,
143 "output_hidden_states": false,
144 "output_scores": false,
145 "pad_token_id": null,
146 "patch_size": 14,
147 "prefix": null,
148 "problem_type": null,
149 "projection_dim": 512,
150 "pruned_heads": {},
151 "remove_invalid_values": false,
152 "repetition_penalty": 1.0,
153 "return_dict": true,
154 "return_dict_in_generate": false,
155 "sep_token_id": null,
156 "task_specific_params": null,
157 "temperature": 1.0,
158 "tf_legacy_loss": false,
159 "tie_encoder_decoder": false,
160 "tie_word_embeddings": true,
161 "tokenizer_class": null,
162 "top_k": 50,
163 "top_p": 1.0,
164 "torch_dtype": null,
165 "torchscript": false,
166 "transformers_version": "4.22.0.dev0",
167 "typical_p": 1.0,
168 "use_bfloat16": false
169 },
170 "vision_config_dict": {
171 "hidden_size": 1024,
172 "intermediate_size": 4096,
173 "num_attention_heads": 16,
174 "num_hidden_layers": 24,
175 "patch_size": 14
176 }
177 }
178