p1atdev commited on
Commit
e606106
·
verified ·
1 Parent(s): b2f00f3

Upload ModernBertForSequenceClassification

Browse files
Files changed (2) hide show
  1. config.json +55 -0
  2. model.safetensors +3 -0
config.json ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "ModernBertForSequenceClassification"
4
+ ],
5
+ "attention_bias": false,
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 1,
8
+ "classifier_activation": "gelu",
9
+ "classifier_bias": false,
10
+ "classifier_dropout": 0.0,
11
+ "classifier_pooling": "cls",
12
+ "cls_token_id": 6,
13
+ "decoder_bias": true,
14
+ "deterministic_flash_attn": false,
15
+ "embedding_dropout": 0.0,
16
+ "eos_token_id": 2,
17
+ "global_attn_every_n_layers": 3,
18
+ "global_rope_theta": 160000.0,
19
+ "gradient_checkpointing": false,
20
+ "hidden_activation": "gelu",
21
+ "hidden_size": 256,
22
+ "id2label": {
23
+ "0": "Safe",
24
+ "1": "Dangerous"
25
+ },
26
+ "initializer_cutoff_factor": 2.0,
27
+ "initializer_range": 0.02,
28
+ "intermediate_size": 1024,
29
+ "label2id": {
30
+ "Dangerous": 1,
31
+ "Safe": 0
32
+ },
33
+ "layer_norm_eps": 1e-05,
34
+ "local_attention": 128,
35
+ "local_rope_theta": 10000.0,
36
+ "max_position_embeddings": 8192,
37
+ "mlp_bias": false,
38
+ "mlp_dropout": 0.0,
39
+ "model_type": "modernbert",
40
+ "norm_bias": false,
41
+ "norm_eps": 1e-05,
42
+ "num_attention_heads": 4,
43
+ "num_hidden_layers": 10,
44
+ "pad_token_id": 3,
45
+ "position_embedding_type": "rope",
46
+ "problem_type": "single_label_classification",
47
+ "reference_compile": false,
48
+ "repad_logits_with_grad": false,
49
+ "sep_token_id": 4,
50
+ "sparse_pred_ignore_index": -100,
51
+ "sparse_prediction": false,
52
+ "torch_dtype": "bfloat16",
53
+ "transformers_version": "4.50.3",
54
+ "vocab_size": 102400
55
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c1b9526abd6c89f9e81dc8db4180eee259180b11b7065b67f90c30e30d0b178f
3
+ size 73550324