From fef8240c23249998c008b69d19db5da4abab3f48 Mon Sep 17 00:00:00 2001 From: Leandro von Werra Date: Wed, 17 Jul 2024 17:32:17 +0200 Subject: [PATCH] fix arg parsing in chat.py (#1846) Co-authored-by: leandro --- examples/scripts/chat.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/examples/scripts/chat.py b/examples/scripts/chat.py index 599cb0ccc1..cda4200d0d 100644 --- a/examples/scripts/chat.py +++ b/examples/scripts/chat.py @@ -21,6 +21,7 @@ import copy import json import os +import sys import pwd import re import time @@ -258,12 +259,11 @@ def parse_eos_tokens(tokenizer, eos_tokens, eos_token_ids): def chat_cli(): parser = TrlParser(ChatArguments) - args = parser.parse_args_into_dataclasses()[0] - if args.config == "default": - args.config = os.path.join(os.path.dirname(__file__), "config/default_chat_config.yaml") - if args.config.lower() == "none": - args.config = None - args = parser.update_dataclasses_with_config([args])[0] + + if "--config" not in sys.argv: + sys.argv.append("--config") + sys.argv.append(os.path.join(os.path.dirname(__file__), "config/default_chat_config.yaml")) + args = parser.parse_args_and_config()[0] if args.examples is None: args.examples = {}