Seperate concerns (PR intel#92 solves it)

aahouzi · Jan 29, 2024 · b13074d · b13074d
1 parent 40d2c84
commit b13074d
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/scripts/inference.py b/scripts/inference.py
@@ -133,9 +133,12 @@ def main(args_in: Optional[List[str]] = None) -> None:
     if is_win():
         path = Path(args.build_dir, "./Bin/Release/run_{}.exe".format(model_name))
     else:
-        import neural_speed
-        package_path = os.path.dirname(neural_speed.__file__)
-        path = Path(package_path, "./run_{}".format(model_name))
+        if args.one_click_run == "True":
+            import neural_speed
+            package_path = os.path.dirname(neural_speed.__file__)
+            path = Path(package_path, "./run_{}".format(model_name))
+        else:
+            path = Path(args.build_dir, "./bin/run_{}".format(model_name))
 
     if not path.exists():
         print("Please build graph first or select the correct model name.")

diff --git a/scripts/quantize.py b/scripts/quantize.py
@@ -103,9 +103,12 @@ def main(args_in: Optional[List[str]] = None) -> None:
     if is_win():
         path = Path(args.build_dir, "./Bin/Release/quant_{}.exe".format(model_name))
     else:
-        import neural_speed
-        package_path = os.path.dirname(neural_speed.__file__)
-        path = Path(package_path, "./quant_{}".format(model_name))
+        if args.one_click_run == "True":
+            import neural_speed
+            package_path = os.path.dirname(neural_speed.__file__)
+            path = Path(package_path, "./quant_{}".format(model_name))
+        else:
+            path = Path(args.build_dir, "./bin/quant_{}".format(model_name))
     if not path.exists():
         print(path)
         print("Please build graph first or select the correct model name.")