From ac6c93df1f6bbf7aa78c68d7e28e2c45614b6f65 Mon Sep 17 00:00:00 2001 From: Injae Ryou Date: Tue, 1 Jul 2025 22:05:13 +0900 Subject: [PATCH] [parser] update config loading to use OmegaConf #7793 (#8505) --- src/llamafactory/hparams/parser.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/llamafactory/hparams/parser.py b/src/llamafactory/hparams/parser.py index 91cee729..54becc5b 100644 --- a/src/llamafactory/hparams/parser.py +++ b/src/llamafactory/hparams/parser.py @@ -62,11 +62,11 @@ def read_args(args: Optional[Union[dict[str, Any], list[str]]] = None) -> Union[ if sys.argv[1].endswith(".yaml") or sys.argv[1].endswith(".yml"): override_config = OmegaConf.from_cli(sys.argv[2:]) - dict_config = yaml.safe_load(Path(sys.argv[1]).absolute().read_text()) + dict_config = OmegaConf.load(Path(sys.argv[1]).absolute()) return OmegaConf.to_container(OmegaConf.merge(dict_config, override_config)) elif sys.argv[1].endswith(".json"): override_config = OmegaConf.from_cli(sys.argv[2:]) - dict_config = json.loads(Path(sys.argv[1]).absolute().read_text()) + dict_config = OmegaConf.load(Path(sys.argv[1]).absolute()) return OmegaConf.to_container(OmegaConf.merge(dict_config, override_config)) else: return sys.argv[1:]