Skip to content

Commit 0e4c901

Browse files
committed
Fixup json load
1 parent 158a672 commit 0e4c901

File tree

1 file changed

+2
-2
lines changed
  • models/turbine_models/custom_models/torchbench

1 file changed

+2
-2
lines changed

models/turbine_models/custom_models/torchbench/export.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -420,7 +420,7 @@ def run_main(model_id, args, tb_dir, tb_args):
420420
if args.compile_to in ["torch", "mlir"]:
421421
safe_name = utils.create_safe_name(
422422
model_id,
423-
f"_{static_dim}_{precision}",
423+
f"_{static_dim}_{args.precision}",
424424
)
425425
with open(f"{safe_name}.mlir", "w+") as f:
426426
f.write(mod_str)
@@ -449,7 +449,7 @@ def run_main(model_id, args, tb_dir, tb_args):
449449
for list in args.model_lists:
450450
torchbench_models_dict = json.load(list)
451451
with open(args.models_json, "r") as f:
452-
torchbench_models_dict = json.load(file)
452+
torchbench_models_dict = json.load(f)
453453

454454
tb_dir = setup_torchbench_cwd()
455455
if args.model_id.lower() == "all":

0 commit comments

Comments
 (0)