423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-14 23:58:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

[parser] update config loading to use OmegaConf #7793 (#8505)

Browse Source
This commit is contained in:
Injae Ryou 2025-07-01 22:05:13 +09:00 committed by GitHub
parent d17a672251
commit a5a93597b1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/llamafactory/hparams/parser.py
View File

@ -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"): if sys.argv[1].endswith(".yaml") or sys.argv[1].endswith(".yml"):
override_config = OmegaConf.from_cli(sys.argv[2:]) 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)) return OmegaConf.to_container(OmegaConf.merge(dict_config, override_config))
elif sys.argv[1].endswith(".json"): elif sys.argv[1].endswith(".json"):
override_config = OmegaConf.from_cli(sys.argv[2:]) 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)) return OmegaConf.to_container(OmegaConf.merge(dict_config, override_config))
else: else:
return sys.argv[1:] return sys.argv[1:]