apunkt/doc-to-lora
1
0
Fork 0
mirror of https://github.com/SakanaAI/doc-to-lora.git synced 2026-07-23 17:01:04 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix yaml load

Browse source
This commit is contained in:
51616 2025-01-14 21:12:32 +00:00
parent d952c54eee
commit 1d9acec4f3
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
hyperlora/watcher.py
View file

@ -84,7 +84,7 @@ if __name__ == "__main__":
run_dir = file.split("/checkpoint")[0]
checkpoint_dir = os.path.dirname(file)
args = argparse.Namespace(
**yaml.safe_load(open(f"{run_dir}/args.yaml", "r"))
**yaml.unsafe_load(open(f"{run_dir}/args.yaml", "r"))
)
args.output_dir = checkpoint_dir
args.logging_dir = checkpoint_dir