Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-18 08:44:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

olive do not ignore venv path

Browse Source
This commit is contained in:
Seunghoon Lee
2024-02-02 14:08:02 +09:00
parent be4534c03f
commit 69599f01e5
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/onnx_impl/__init__.py
+2 -1
View File
@@ -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