This commit is contained in:
Goekdeniz-Guelmez
2025-02-05 09:48:00 +01:00
parent d84ad0cf86
commit a33cad84b4
2 changed files with 25 additions and 66 deletions

View File

@@ -206,15 +206,15 @@ def build_parser():
)
parser.add_argument(
"--use-chat-template",
type=bool,
action="store_true",
help="If the model is a Chat model, use the Chat template.",
default=False,
default=None,
)
parser.add_argument(
"--use-prompt",
type=bool,
help="Rather to use the prompt from teh R1 paper.",
default=False,
action="store_true",
help="Rather to use the prompt from the R1 paper.",
default=None,
)
return parser