From 0e4c901cb5acdd4ce7248850da30d3bc5e3173a0 Mon Sep 17 00:00:00 2001 From: Ean Garvey Date: Thu, 17 Oct 2024 11:04:02 -0500 Subject: [PATCH] Fixup json load --- models/turbine_models/custom_models/torchbench/export.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/models/turbine_models/custom_models/torchbench/export.py b/models/turbine_models/custom_models/torchbench/export.py index 84eef392..3f145224 100644 --- a/models/turbine_models/custom_models/torchbench/export.py +++ b/models/turbine_models/custom_models/torchbench/export.py @@ -420,7 +420,7 @@ def run_main(model_id, args, tb_dir, tb_args): if args.compile_to in ["torch", "mlir"]: safe_name = utils.create_safe_name( model_id, - f"_{static_dim}_{precision}", + f"_{static_dim}_{args.precision}", ) with open(f"{safe_name}.mlir", "w+") as f: f.write(mod_str) @@ -449,7 +449,7 @@ def run_main(model_id, args, tb_dir, tb_args): for list in args.model_lists: torchbench_models_dict = json.load(list) with open(args.models_json, "r") as f: - torchbench_models_dict = json.load(file) + torchbench_models_dict = json.load(f) tb_dir = setup_torchbench_cwd() if args.model_id.lower() == "all":