tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Fix load error

Browse Source
This commit is contained in:
NanoCode012
2023-05-28 02:55:46 +09:00
parent bdfe7c9201
commit 93acb648bd
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/finetune.py
View File

@@ -131,10 +131,10 @@ def train(
# load the config from the yaml file # load the config from the yaml file
with open(config, "r") as f: with open(config, "r") as f:
cfg: Dict = Dict(lambda: None, yaml.load(f, Loader=yaml.Loader)) cfg: Dict = Dict(yaml.load(f, Loader=yaml.Loader))
# if there are any options passed in the cli, if it is something that seems valid from the yaml, # if there are any options passed in the cli, if it is something that seems valid from the yaml,
# then overwrite the value # then overwrite the value
cfg_keys = dict(cfg).keys() cfg_keys = cfg.keys()
for k in kwargs: for k in kwargs:
# if not strict, allow writing to cfg even if it's not in the yml already # if not strict, allow writing to cfg even if it's not in the yml already
if k in cfg_keys or cfg.strict is False: if k in cfg_keys or cfg.strict is False: