1580158
/
lora-scripts
mirror of https://github.com/Akegarasu/lora-scripts
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

skip onnxruntime install by default

pull/608/head
Akegarasu 2024-12-17 11:56:50 +08:00
parent 950d4fdbda
commit 935f78fe20
No known key found for this signature in database
GPG Key ID: DACA951FEBA569A2
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mikazuki/launch_utils.py
View File

@ -259,7 +259,7 @@ def check_run(file: str) -> bool:
return result.returncode == 0
def prepare_environment(disable_auto_mirror: bool = True, skip_prepare_onnxruntime: bool = False):
def prepare_environment(disable_auto_mirror: bool = True):
if sys.platform == "win32":
# disable triton on windows
os.environ["XFORMERS_FORCE_DISABLE_TRITON"] = "1"
@ -287,8 +287,9 @@ def prepare_environment(disable_auto_mirror: bool = True, skip_prepare_onnxrunti
validate_requirements("requirements.txt")
setup_windows_bitsandbytes()
if not skip_prepare_onnxruntime:
setup_onnxruntime()
# if not skip_prepare_onnxruntime:
# setup_onnxruntime()
def catch_exception(f):