Update config.json
Browse files- config.json +3 -9
config.json
CHANGED
|
@@ -1,17 +1,13 @@
|
|
| 1 |
{
|
| 2 |
"_name_or_path": "TheBlueScrubs/ModernBERT-base-TBS-MedicalRelevance",
|
| 3 |
-
"architectures": [
|
| 4 |
-
"ModernBertForSequenceClassification"
|
| 5 |
-
],
|
| 6 |
"model_type": "modernbert",
|
| 7 |
|
| 8 |
-
/* --- task-specific additions --- */
|
| 9 |
"num_labels": 1,
|
| 10 |
"problem_type": "regression",
|
| 11 |
-
"id2label": {
|
| 12 |
-
"label2id": {
|
| 13 |
|
| 14 |
-
/* --- original backbone settings (unchanged) --- */
|
| 15 |
"hidden_size": 768,
|
| 16 |
"intermediate_size": 1152,
|
| 17 |
"num_hidden_layers": 22,
|
|
@@ -43,14 +39,12 @@
|
|
| 43 |
"sparse_pred_ignore_index": -100,
|
| 44 |
"torch_dtype": "float32",
|
| 45 |
|
| 46 |
-
/* --- special tokens (unchanged) --- */
|
| 47 |
"bos_token_id": 50281,
|
| 48 |
"eos_token_id": 50282,
|
| 49 |
"cls_token_id": 50281,
|
| 50 |
"sep_token_id": 50282,
|
| 51 |
"pad_token_id": 50283,
|
| 52 |
|
| 53 |
-
/* --- misc --- */
|
| 54 |
"position_embedding_type": "absolute",
|
| 55 |
"reference_compile": null,
|
| 56 |
"transformers_version": "4.48.3",
|
|
|
|
| 1 |
{
|
| 2 |
"_name_or_path": "TheBlueScrubs/ModernBERT-base-TBS-MedicalRelevance",
|
| 3 |
+
"architectures": ["ModernBertForSequenceClassification"],
|
|
|
|
|
|
|
| 4 |
"model_type": "modernbert",
|
| 5 |
|
|
|
|
| 6 |
"num_labels": 1,
|
| 7 |
"problem_type": "regression",
|
| 8 |
+
"id2label": {"0": "relevance"},
|
| 9 |
+
"label2id": {"relevance": 0},
|
| 10 |
|
|
|
|
| 11 |
"hidden_size": 768,
|
| 12 |
"intermediate_size": 1152,
|
| 13 |
"num_hidden_layers": 22,
|
|
|
|
| 39 |
"sparse_pred_ignore_index": -100,
|
| 40 |
"torch_dtype": "float32",
|
| 41 |
|
|
|
|
| 42 |
"bos_token_id": 50281,
|
| 43 |
"eos_token_id": 50282,
|
| 44 |
"cls_token_id": 50281,
|
| 45 |
"sep_token_id": 50282,
|
| 46 |
"pad_token_id": 50283,
|
| 47 |
|
|
|
|
| 48 |
"position_embedding_type": "absolute",
|
| 49 |
"reference_compile": null,
|
| 50 |
"transformers_version": "4.48.3",
|