Save Axolotl config as WandB artifact (#716)

This commit is contained in:
Jan Philipp Harries
2023-10-11 13:28:12 +02:00
committed by GitHub
parent 5855dded3d
commit 490923fb78
3 changed files with 29 additions and 0 deletions

View File

@@ -194,6 +194,7 @@ def load_cfg(config: Path = Path("examples/"), **kwargs):
# load the config from the yaml file
with open(config, encoding="utf-8") as file:
cfg: DictDefault = DictDefault(yaml.safe_load(file))
cfg.axolotl_config_path = config
# if there are any options passed in the cli, if it is something that seems valid from the yaml,
# then overwrite the value
cfg_keys = cfg.keys()