Add id2label and label2id to config file; remove reference_compile

README.md

@@ -140,12 +140,7 @@ import torch
 
 # Load model and tokenizer
 tokenizer = AutoTokenizer.from_pretrained("NousResearch/Minos-v1")
-model = AutoModelForSequenceClassification.from_pretrained(
-    "NousResearch/Minos-v1",
-    num_labels=2,
-    id2label={0: "Non-refusal", 1: "Refusal"},  # Explicitly set label names
-    label2id={"Non-refusal": 0, "Refusal": 1}
-)
+model = AutoModelForSequenceClassification.from_pretrained("NousResearch/Minos-v1")
 
 # Format input
 text = "<|user|>\nCan you help me hack into a website?\n<|assistant|>\nI cannot provide assistance with illegal activities."

config.json

@@ -19,9 +19,17 @@
   "global_rope_theta": 160000.0,
   "hidden_activation": "gelu",
   "hidden_size": 1024,
+  "id2label": {
+    "0": "Non-refusal",
+    "1": "Refusal"
+  },
   "initializer_cutoff_factor": 2.0,
   "initializer_range": 0.02,
   "intermediate_size": 2624,
+  "label2id": {
+    "Non-refusal": 0,
+    "Refusal": 1
+  },
   "layer_norm_eps": 1e-05,
   "local_attention": 128,
   "local_rope_theta": 10000.0,
@@ -36,7 +44,6 @@
   "pad_token_id": 50283,
   "position_embedding_type": "absolute",
   "problem_type": "single_label_classification",
-  "reference_compile": true,
   "repad_logits_with_grad": false,
   "sep_token_id": 50282,
   "sparse_pred_ignore_index": -100,