Hugging Face's logo

Intel
/
toxic-prompt-roberta

Text Classification
Transformers
ONNX
Safetensors
Habana
roberta
Inference Endpoints
Model card Files Community
4

Initial model commit

#1
by daniel-de-leon - opened
base: refs/heads/main
←
from: refs/pr/1
Discussion Files changed
+301217
-0
Files changed (20) hide show
  1. config.json +36 -0
  2. gaudi_config.json +10 -0
  3. merges.txt +0 -0
  4. model.safetensors +3 -0
  5. onnx/config.json +27 -0
  6. onnx/merges.txt +0 -0
  7. onnx/model.onnx +3 -0
  8. onnx/special_tokens_map.json +51 -0
  9. onnx/tokenizer.json +0 -0
  10. onnx/tokenizer_config.json +57 -0
  11. onnx/vocab.json +0 -0
  12. optimizer.pt +3 -0
  13. rng_state.pth +3 -0
  14. scheduler.pt +3 -0
  15. special_tokens_map.json +15 -0
  16. tokenizer.json +0 -0
  17. tokenizer_config.json +57 -0
  18. trainer_state.json +228 -0
  19. training_args.bin +3 -0
  20. vocab.json +0 -0
config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "roberta-base",
3
+ "architectures": [
4
+ "RobertaForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "bos_token_id": 0,
8
+ "classifier_dropout": null,
9
+ "eos_token_id": 2,
10
+ "hidden_act": "gelu",
11
+ "hidden_dropout_prob": 0.1,
12
+ "hidden_size": 768,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 3072,
15
+ "layer_norm_eps": 1e-05,
16
+ "max_position_embeddings": 514,
17
+ "model_type": "roberta",
18
+ "num_attention_heads": 12,
19
+ "num_hidden_layers": 12,
20
+ "pad_token_id": 1,
21
+ "position_embedding_type": "absolute",
22
+ "problem_type": "single_label_classification",
23
+ "torch_dtype": "float32",
24
+ "transformers_version": "4.40.2",
25
+ "type_vocab_size": 1,
26
+ "use_cache": true,
27
+ "vocab_size": 50265,
28
+ "id2label": {
29
+ "0": "NOT_TOXIC",
30
+ "1": "TOXIC"
31
+ },
32
+ "label2id": {
33
+ "NOT_TOXIC": 0,
34
+ "TOXIC": 1
35
+ }
36
+ }
gaudi_config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "autocast_bf16_ops": null,
3
+ "autocast_fp32_ops": null,
4
+ "optimum_version": "1.21.2",
5
+ "transformers_version": "4.40.2",
6
+ "use_dynamic_shapes": false,
7
+ "use_fused_adam": true,
8
+ "use_fused_clip_norm": true,
9
+ "use_torch_autocast": true
10
+ }
merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:46d7386384399fe685d6e3d189d4fb03bade10c9739d545cd4098213ae94d432
3
+ size 498612824
onnx/config.json ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "testmg/test-toxic-roberta",
3
+ "architectures": [
4
+ "RobertaForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "bos_token_id": 0,
8
+ "classifier_dropout": null,
9
+ "eos_token_id": 2,
10
+ "hidden_act": "gelu",
11
+ "hidden_dropout_prob": 0.1,
12
+ "hidden_size": 768,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 3072,
15
+ "layer_norm_eps": 1e-05,
16
+ "max_position_embeddings": 514,
17
+ "model_type": "roberta",
18
+ "num_attention_heads": 12,
19
+ "num_hidden_layers": 12,
20
+ "pad_token_id": 1,
21
+ "position_embedding_type": "absolute",
22
+ "problem_type": "single_label_classification",
23
+ "transformers_version": "4.43.4",
24
+ "type_vocab_size": 1,
25
+ "use_cache": true,
26
+ "vocab_size": 50265
27
+ }
onnx/merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
onnx/model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:280036fe7b9f0909d264137f34a454cd377cd4a37a457f68bc1cd93344f4cb73
3
+ size 498870601
onnx/special_tokens_map.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<s>",
4
+ "lstrip": false,
5
+ "normalized": true,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "cls_token": {
10
+ "content": "<s>",
11
+ "lstrip": false,
12
+ "normalized": true,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "eos_token": {
17
+ "content": "</s>",
18
+ "lstrip": false,
19
+ "normalized": true,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "mask_token": {
24
+ "content": "<mask>",
25
+ "lstrip": true,
26
+ "normalized": false,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ },
30
+ "pad_token": {
31
+ "content": "<pad>",
32
+ "lstrip": false,
33
+ "normalized": true,
34
+ "rstrip": false,
35
+ "single_word": false
36
+ },
37
+ "sep_token": {
38
+ "content": "</s>",
39
+ "lstrip": false,
40
+ "normalized": true,
41
+ "rstrip": false,
42
+ "single_word": false
43
+ },
44
+ "unk_token": {
45
+ "content": "<unk>",
46
+ "lstrip": false,
47
+ "normalized": true,
48
+ "rstrip": false,
49
+ "single_word": false
50
+ }
51
+ }
onnx/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
onnx/tokenizer_config.json ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "added_tokens_decoder": {
4
+ "0": {
5
+ "content": "<s>",
6
+ "lstrip": false,
7
+ "normalized": true,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
+ "1": {
13
+ "content": "<pad>",
14
+ "lstrip": false,
15
+ "normalized": true,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ },
20
+ "2": {
21
+ "content": "</s>",
22
+ "lstrip": false,
23
+ "normalized": true,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": true
27
+ },
28
+ "3": {
29
+ "content": "<unk>",
30
+ "lstrip": false,
31
+ "normalized": true,
32
+ "rstrip": false,
33
+ "single_word": false,
34
+ "special": true
35
+ },
36
+ "50264": {
37
+ "content": "<mask>",
38
+ "lstrip": true,
39
+ "normalized": false,
40
+ "rstrip": false,
41
+ "single_word": false,
42
+ "special": true
43
+ }
44
+ },
45
+ "bos_token": "<s>",
46
+ "clean_up_tokenization_spaces": true,
47
+ "cls_token": "<s>",
48
+ "eos_token": "</s>",
49
+ "errors": "replace",
50
+ "mask_token": "<mask>",
51
+ "model_max_length": 512,
52
+ "pad_token": "<pad>",
53
+ "sep_token": "</s>",
54
+ "tokenizer_class": "RobertaTokenizer",
55
+ "trim_offsets": true,
56
+ "unk_token": "<unk>"
57
+ }
onnx/vocab.json ADDED
The diff for this file is too large to render. See raw diff
 
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b6f0c256603715309937a4ac5ea2fe9c25cd31e5fdd29a523bc6066d1e0d780c
3
+ size 997277306
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c50c01182c5ba44de7f7e08adaaed4e3a198e9f7e4230f6915be7d54ac67cd50
3
+ size 18020
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:108df313bb6a5d5592937dc0afbe315665923758d97c08b83089e679470ac9e2
3
+ size 1000
special_tokens_map.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": "<s>",
3
+ "cls_token": "<s>",
4
+ "eos_token": "</s>",
5
+ "mask_token": {
6
+ "content": "<mask>",
7
+ "lstrip": true,
8
+ "normalized": false,
9
+ "rstrip": false,
10
+ "single_word": false
11
+ },
12
+ "pad_token": "<pad>",
13
+ "sep_token": "</s>",
14
+ "unk_token": "<unk>"
15
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "added_tokens_decoder": {
4
+ "0": {
5
+ "content": "<s>",
6
+ "lstrip": false,
7
+ "normalized": true,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
+ "1": {
13
+ "content": "<pad>",
14
+ "lstrip": false,
15
+ "normalized": true,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ },
20
+ "2": {
21
+ "content": "</s>",
22
+ "lstrip": false,
23
+ "normalized": true,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": true
27
+ },
28
+ "3": {
29
+ "content": "<unk>",
30
+ "lstrip": false,
31
+ "normalized": true,
32
+ "rstrip": false,
33
+ "single_word": false,
34
+ "special": true
35
+ },
36
+ "50264": {
37
+ "content": "<mask>",
38
+ "lstrip": true,
39
+ "normalized": false,
40
+ "rstrip": false,
41
+ "single_word": false,
42
+ "special": true
43
+ }
44
+ },
45
+ "bos_token": "<s>",
46
+ "clean_up_tokenization_spaces": true,
47
+ "cls_token": "<s>",
48
+ "eos_token": "</s>",
49
+ "errors": "replace",
50
+ "mask_token": "<mask>",
51
+ "model_max_length": 512,
52
+ "pad_token": "<pad>",
53
+ "sep_token": "</s>",
54
+ "tokenizer_class": "RobertaTokenizer",
55
+ "trim_offsets": true,
56
+ "unk_token": "<unk>"
57
+ }
trainer_state.json ADDED
@@ -0,0 +1,228 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 3.0,
5
+ "eval_steps": 40,
6
+ "global_step": 384,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.3125,
13
+ "grad_norm": 2.4588866233825684,
14
+ "learning_rate": 6.578947368421053e-06,
15
+ "loss": 0.3311,
16
+ "max_memory_allocated (GB)": 13.36,
17
+ "memory_allocated (GB)": 11.92,
18
+ "step": 40,
19
+ "total_memory_available (GB)": 94.62
20
+ },
21
+ {
22
+ "epoch": 0.3125,
23
+ "eval_accuracy": 0.9449360865290068,
24
+ "eval_auroc": 0.9216382070437567,
25
+ "eval_loss": 0.16689223051071167,
26
+ "eval_runtime": 4.7738,
27
+ "eval_samples_per_second": 213.038,
28
+ "eval_steps_per_second": 6.703,
29
+ "max_memory_allocated (GB)": 13.36,
30
+ "memory_allocated (GB)": 11.92,
31
+ "step": 40,
32
+ "total_memory_available (GB)": 94.62
33
+ },
34
+ {
35
+ "epoch": 0.625,
36
+ "grad_norm": 6.371248245239258,
37
+ "learning_rate": 1.3157894736842106e-05,
38
+ "loss": 0.1336,
39
+ "max_memory_allocated (GB)": 13.36,
40
+ "memory_allocated (GB)": 11.92,
41
+ "step": 80,
42
+ "total_memory_available (GB)": 94.62
43
+ },
44
+ {
45
+ "epoch": 0.625,
46
+ "eval_accuracy": 0.951819075712881,
47
+ "eval_auroc": 0.9588580576307363,
48
+ "eval_loss": 0.14572063088417053,
49
+ "eval_runtime": 1.6415,
50
+ "eval_samples_per_second": 619.571,
51
+ "eval_steps_per_second": 19.495,
52
+ "max_memory_allocated (GB)": 13.36,
53
+ "memory_allocated (GB)": 11.92,
54
+ "step": 80,
55
+ "total_memory_available (GB)": 94.62
56
+ },
57
+ {
58
+ "epoch": 0.9375,
59
+ "grad_norm": 0.8174734115600586,
60
+ "learning_rate": 1.9736842105263158e-05,
61
+ "loss": 0.092,
62
+ "max_memory_allocated (GB)": 13.36,
63
+ "memory_allocated (GB)": 11.92,
64
+ "step": 120,
65
+ "total_memory_available (GB)": 94.62
66
+ },
67
+ {
68
+ "epoch": 0.9375,
69
+ "eval_accuracy": 0.9587020648967551,
70
+ "eval_auroc": 0.9663153681963713,
71
+ "eval_loss": 0.13402177393436432,
72
+ "eval_runtime": 2.0205,
73
+ "eval_samples_per_second": 503.333,
74
+ "eval_steps_per_second": 15.837,
75
+ "max_memory_allocated (GB)": 13.36,
76
+ "memory_allocated (GB)": 11.92,
77
+ "step": 120,
78
+ "total_memory_available (GB)": 94.62
79
+ },
80
+ {
81
+ "epoch": 1.25,
82
+ "grad_norm": 10.359611511230469,
83
+ "learning_rate": 2.6315789473684212e-05,
84
+ "loss": 0.0683,
85
+ "max_memory_allocated (GB)": 13.36,
86
+ "memory_allocated (GB)": 11.92,
87
+ "step": 160,
88
+ "total_memory_available (GB)": 94.62
89
+ },
90
+ {
91
+ "epoch": 1.25,
92
+ "eval_accuracy": 0.9616519174041298,
93
+ "eval_auroc": 0.9728054962646745,
94
+ "eval_loss": 0.15004919469356537,
95
+ "eval_runtime": 1.9839,
96
+ "eval_samples_per_second": 512.639,
97
+ "eval_steps_per_second": 16.13,
98
+ "max_memory_allocated (GB)": 13.36,
99
+ "memory_allocated (GB)": 11.92,
100
+ "step": 160,
101
+ "total_memory_available (GB)": 94.62
102
+ },
103
+ {
104
+ "epoch": 1.5625,
105
+ "grad_norm": 0.29452481865882874,
106
+ "learning_rate": 3.289473684210527e-05,
107
+ "loss": 0.0778,
108
+ "max_memory_allocated (GB)": 13.36,
109
+ "memory_allocated (GB)": 11.92,
110
+ "step": 200,
111
+ "total_memory_available (GB)": 94.62
112
+ },
113
+ {
114
+ "epoch": 1.5625,
115
+ "eval_accuracy": 0.9567354965585054,
116
+ "eval_auroc": 0.9560165421558164,
117
+ "eval_loss": 0.19474196434020996,
118
+ "eval_runtime": 1.8404,
119
+ "eval_samples_per_second": 552.609,
120
+ "eval_steps_per_second": 17.388,
121
+ "max_memory_allocated (GB)": 13.36,
122
+ "memory_allocated (GB)": 11.92,
123
+ "step": 200,
124
+ "total_memory_available (GB)": 94.62
125
+ },
126
+ {
127
+ "epoch": 1.875,
128
+ "grad_norm": 0.4737164378166199,
129
+ "learning_rate": 3.9473684210526316e-05,
130
+ "loss": 0.1011,
131
+ "max_memory_allocated (GB)": 13.36,
132
+ "memory_allocated (GB)": 11.92,
133
+ "step": 240,
134
+ "total_memory_available (GB)": 94.62
135
+ },
136
+ {
137
+ "epoch": 1.875,
138
+ "eval_accuracy": 0.967551622418879,
139
+ "eval_auroc": 0.9674493062966916,
140
+ "eval_loss": 0.13491444289684296,
141
+ "eval_runtime": 1.7786,
142
+ "eval_samples_per_second": 571.791,
143
+ "eval_steps_per_second": 17.991,
144
+ "max_memory_allocated (GB)": 13.36,
145
+ "memory_allocated (GB)": 11.92,
146
+ "step": 240,
147
+ "total_memory_available (GB)": 94.62
148
+ },
149
+ {
150
+ "epoch": 2.1875,
151
+ "grad_norm": 0.6345912218093872,
152
+ "learning_rate": 4.605263157894737e-05,
153
+ "loss": 0.06,
154
+ "max_memory_allocated (GB)": 13.36,
155
+ "memory_allocated (GB)": 11.92,
156
+ "step": 280,
157
+ "total_memory_available (GB)": 94.62
158
+ },
159
+ {
160
+ "epoch": 2.1875,
161
+ "eval_accuracy": 0.9655850540806293,
162
+ "eval_auroc": 0.9698305763073639,
163
+ "eval_loss": 0.1612750142812729,
164
+ "eval_runtime": 1.893,
165
+ "eval_samples_per_second": 537.248,
166
+ "eval_steps_per_second": 16.905,
167
+ "max_memory_allocated (GB)": 13.36,
168
+ "memory_allocated (GB)": 11.92,
169
+ "step": 280,
170
+ "total_memory_available (GB)": 94.62
171
+ },
172
+ {
173
+ "epoch": 2.5,
174
+ "grad_norm": 13.752758026123047,
175
+ "learning_rate": 4e-05,
176
+ "loss": 0.0508,
177
+ "max_memory_allocated (GB)": 13.36,
178
+ "memory_allocated (GB)": 11.92,
179
+ "step": 320,
180
+ "total_memory_available (GB)": 94.62
181
+ },
182
+ {
183
+ "epoch": 2.5,
184
+ "eval_accuracy": 0.9685349065880039,
185
+ "eval_auroc": 0.9268209711846318,
186
+ "eval_loss": 0.19579041004180908,
187
+ "eval_runtime": 1.7675,
188
+ "eval_samples_per_second": 575.401,
189
+ "eval_steps_per_second": 18.105,
190
+ "max_memory_allocated (GB)": 13.36,
191
+ "memory_allocated (GB)": 11.92,
192
+ "step": 320,
193
+ "total_memory_available (GB)": 94.62
194
+ },
195
+ {
196
+ "epoch": 2.8125,
197
+ "grad_norm": 15.928082466125488,
198
+ "learning_rate": 1.5e-05,
199
+ "loss": 0.0722,
200
+ "max_memory_allocated (GB)": 13.36,
201
+ "memory_allocated (GB)": 11.92,
202
+ "step": 360,
203
+ "total_memory_available (GB)": 94.62
204
+ },
205
+ {
206
+ "epoch": 2.8125,
207
+ "eval_accuracy": 0.967551622418879,
208
+ "eval_auroc": 0.9649679829242261,
209
+ "eval_loss": 0.15064473450183868,
210
+ "eval_runtime": 1.906,
211
+ "eval_samples_per_second": 533.581,
212
+ "eval_steps_per_second": 16.789,
213
+ "max_memory_allocated (GB)": 13.36,
214
+ "memory_allocated (GB)": 11.92,
215
+ "step": 360,
216
+ "total_memory_available (GB)": 94.62
217
+ }
218
+ ],
219
+ "logging_steps": 40,
220
+ "max_steps": 384,
221
+ "num_input_tokens_seen": 0,
222
+ "num_train_epochs": 3,
223
+ "save_steps": 500,
224
+ "total_flos": 3233108648263680.0,
225
+ "train_batch_size": 32,
226
+ "trial_name": null,
227
+ "trial_params": null
228
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:060db504522c4577307b25bfbb3e6cafaea4180c7c87aa1089d0069c3444a6e4
3
+ size 4856
vocab.json ADDED
The diff for this file is too large to render. See raw diff