From 56132983cfe4c026cf37f01b8af4d6568289c85d Mon Sep 17 00:00:00 2001 From: hiyouga <467089858@qq.com> Date: Sun, 6 Oct 2024 10:34:55 +0800 Subject: [PATCH] fix #5611 Former-commit-id: a45f3f5461e2936b9e119eda2ef4d8c7a4131740 --- src/llamafactory/hparams/parser.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/llamafactory/hparams/parser.py b/src/llamafactory/hparams/parser.py index fd112607..38ac87fe 100644 --- a/src/llamafactory/hparams/parser.py +++ b/src/llamafactory/hparams/parser.py @@ -57,7 +57,7 @@ def _parse_args(parser: "HfArgumentParser", args: Optional[Dict[str, Any]] = Non if args is not None: return parser.parse_dict(args) - if len(sys.argv) == 2 and sys.argv[1].endswith(".yaml"): + if len(sys.argv) == 2 and (sys.argv[1].endswith(".yaml") or sys.argv[1].endswith(".yml")): return parser.parse_yaml_file(os.path.abspath(sys.argv[1])) if len(sys.argv) == 2 and sys.argv[1].endswith(".json"):