Flag correction

arcee-ai · Oct 6, 2023 · f8d824e · f8d824e
1 parent 107b9ca
commit f8d824e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dalm/eval/eval_retriever_only.py b/dalm/eval/eval_retriever_only.py
@@ -94,7 +94,7 @@ def parse_args() -> Namespace:
         help="Whether the model is autoregressive or not",
     )
     parser.add_argument(
-        "extract_eos_only",
+        "--extract_eos_only",
         action="store_true",
         help="Whether to extract only the last token of the sequence",
     )