From 69599f01e5ddb640d7d5d02e9519a4dde86ba38f Mon Sep 17 00:00:00 2001 From: Seunghoon Lee Date: Fri, 2 Feb 2024 14:08:02 +0900 Subject: [PATCH] olive do not ignore venv path --- modules/onnx_impl/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/modules/onnx_impl/__init__.py b/modules/onnx_impl/__init__.py index 8b8fdef14..e49858636 100644 --- a/modules/onnx_impl/__init__.py +++ b/modules/onnx_impl/__init__.py @@ -188,9 +188,10 @@ def initialize_olive(): import sys import importlib orig_sys_path = sys.path + venv_dir = os.environ.get("VENV_DIR", os.path.join(os.getcwd(), 'venv')) try: spec = importlib.util.find_spec('onnxruntime.transformers') - sys.path = [d for d in spec.submodule_search_locations + sys.path if sys.path[1] not in d] + sys.path = [d for d in spec.submodule_search_locations + sys.path if sys.path[1] not in d or venv_dir in d] from onnxruntime.transformers import convert_generation # pylint: disable=unused-import spec = importlib.util.find_spec('olive') sys.path = spec.submodule_search_locations + sys.path