notlath commited on
Commit
7202789
·
verified ·
1 Parent(s): dd1e245

Upload ModernBertForSequenceClassification

Browse files
Files changed (2) hide show
  1. config.json +61 -61
  2. model.safetensors +1 -1
config.json CHANGED
@@ -1,61 +1,61 @@
1
- {
2
- "architectures": [
3
- "ModernBertForSequenceClassification"
4
- ],
5
- "attention_bias": false,
6
- "attention_dropout": 0.0,
7
- "bos_token_id": 50281,
8
- "classifier_activation": "silu",
9
- "classifier_bias": false,
10
- "classifier_dropout": 0.0,
11
- "classifier_pooling": "mean",
12
- "cls_token_id": 50281,
13
- "decoder_bias": true,
14
- "deterministic_flash_attn": false,
15
- "dtype": "float32",
16
- "embedding_dropout": 0.0,
17
- "eos_token_id": 50282,
18
- "global_attn_every_n_layers": 3,
19
- "global_rope_theta": 160000.0,
20
- "gradient_checkpointing": false,
21
- "hidden_activation": "gelu",
22
- "hidden_size": 768,
23
- "id2label": {
24
- "0": "Pneumonia",
25
- "1": "Measles",
26
- "2": "Diarrhea",
27
- "3": "Dengue",
28
- "4": "Influenza",
29
- "5": "Typhoid"
30
- },
31
- "initializer_cutoff_factor": 2.0,
32
- "initializer_range": 0.02,
33
- "intermediate_size": 1152,
34
- "label2id": {
35
- "Dengue": 3,
36
- "Diarrhea": 2,
37
- "Influenza": 4,
38
- "Measles": 1,
39
- "Pneumonia": 0,
40
- "Typhoid": 5
41
- },
42
- "layer_norm_eps": 1e-05,
43
- "local_attention": 128,
44
- "local_rope_theta": 10000.0,
45
- "max_position_embeddings": 8192,
46
- "mlp_bias": false,
47
- "mlp_dropout": 0.0,
48
- "model_type": "modernbert",
49
- "norm_bias": false,
50
- "norm_eps": 1e-05,
51
- "num_attention_heads": 12,
52
- "num_hidden_layers": 22,
53
- "pad_token_id": 50283,
54
- "position_embedding_type": "absolute",
55
- "repad_logits_with_grad": false,
56
- "sep_token_id": 50282,
57
- "sparse_pred_ignore_index": -100,
58
- "sparse_prediction": false,
59
- "transformers_version": "4.56.2",
60
- "vocab_size": 50368
61
- }
 
1
+ {
2
+ "architectures": [
3
+ "ModernBertForSequenceClassification"
4
+ ],
5
+ "attention_bias": false,
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 50281,
8
+ "classifier_activation": "silu",
9
+ "classifier_bias": false,
10
+ "classifier_dropout": 0.0,
11
+ "classifier_pooling": "mean",
12
+ "cls_token_id": 50281,
13
+ "decoder_bias": true,
14
+ "deterministic_flash_attn": false,
15
+ "dtype": "float32",
16
+ "embedding_dropout": 0.0,
17
+ "eos_token_id": 50282,
18
+ "global_attn_every_n_layers": 3,
19
+ "global_rope_theta": 160000.0,
20
+ "gradient_checkpointing": false,
21
+ "hidden_activation": "gelu",
22
+ "hidden_size": 768,
23
+ "id2label": {
24
+ "0": "Dengue",
25
+ "1": "Influenza",
26
+ "2": "Typhoid",
27
+ "3": "Measles",
28
+ "4": "Diarrhea",
29
+ "5": "Pneumonia"
30
+ },
31
+ "initializer_cutoff_factor": 2.0,
32
+ "initializer_range": 0.02,
33
+ "intermediate_size": 1152,
34
+ "label2id": {
35
+ "Dengue": 0,
36
+ "Diarrhea": 4,
37
+ "Influenza": 1,
38
+ "Measles": 3,
39
+ "Pneumonia": 5,
40
+ "Typhoid": 2
41
+ },
42
+ "layer_norm_eps": 1e-05,
43
+ "local_attention": 128,
44
+ "local_rope_theta": 10000.0,
45
+ "max_position_embeddings": 8192,
46
+ "mlp_bias": false,
47
+ "mlp_dropout": 0.0,
48
+ "model_type": "modernbert",
49
+ "norm_bias": false,
50
+ "norm_eps": 1e-05,
51
+ "num_attention_heads": 12,
52
+ "num_hidden_layers": 22,
53
+ "pad_token_id": 50283,
54
+ "position_embedding_type": "absolute",
55
+ "repad_logits_with_grad": false,
56
+ "sep_token_id": 50282,
57
+ "sparse_pred_ignore_index": -100,
58
+ "sparse_prediction": false,
59
+ "transformers_version": "4.57.1",
60
+ "vocab_size": 50368
61
+ }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:10881c77df8351f90149bba2382d20833eb4edd75ed70db91c25dd1b3943b279
3
  size 598452088
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:124d013f305accd80c8e24918dfc4630fe324e51779c61fdf758072cdeafa58d
3
  size 598452088