typo

pytorch · Apr 15, 2024 · b544177 · b544177
1 parent 3f3ef41
commit b544177
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/eval.py b/eval.py
@@ -254,13 +254,13 @@ def main(args) -> None:
         max_seq_length,
     )
     print(f"Time to run eval: {time.time() - t1:.02f} seconds.")
-    if dso_path:
+    if builder_args.dso_path:
         print(f"For model {builder_args.dso_path}")
-    elif pte_path:
+    elif builder_args.pte_path:
         print(f"For model {builder_args.pte_path}")
-    elif checkpoint_path:
+    elif builder_args.checkpoint_path:
         print(f"For model {builder_args.checkpoint_path}")
-    elif checkpoint_dir:
+    elif builder_args.checkpoint_dir:
         print(f"For model {builder_args.checkpoint_dir}")
     else:
         raise RuntimeError("Well That's Fine. How did we get here")