From 1d9acec4f36717e40586d36acea2c9fae2882208 Mon Sep 17 00:00:00 2001 From: 51616 Date: Tue, 14 Jan 2025 21:12:32 +0000 Subject: [PATCH] fix yaml load --- hyperlora/watcher.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/hyperlora/watcher.py b/hyperlora/watcher.py index 2985f34..4e0d95e 100644 --- a/hyperlora/watcher.py +++ b/hyperlora/watcher.py @@ -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