Install pre-compiled wheels for Linux (#74)

This commit is contained in:
Sam 2023-05-31 13:41:54 -04:00 committed by GitHub
parent be98e74337
commit 5405635305
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -164,14 +164,18 @@ def update_dependencies():
# If the path does not exist, then the install failed # If the path does not exist, then the install failed
quant_cuda_path_regex = os.path.join(site_packages_path, "quant_cuda*/") quant_cuda_path_regex = os.path.join(site_packages_path, "quant_cuda*/")
if not glob.glob(quant_cuda_path_regex): if not glob.glob(quant_cuda_path_regex):
# Attempt installation via alternative, Windows-specific method # Attempt installation via alternative, Windows/Linux-specific method
if sys.platform.startswith("win"): if sys.platform.startswith("win") or sys.platform.startswith("linux"):
print("\n\n*******************************************************************") print("\n\n*******************************************************************")
print("* WARNING: GPTQ-for-LLaMa compilation failed, but this is FINE and can be ignored!") print("* WARNING: GPTQ-for-LLaMa compilation failed, but this is FINE and can be ignored!")
print("* The installer will proceed to install a pre-compiled wheel.") print("* The installer will proceed to install a pre-compiled wheel.")
print("*******************************************************************\n\n") print("*******************************************************************\n\n")
result = run_cmd("python -m pip install https://github.com/jllllll/GPTQ-for-LLaMa-Wheels/raw/main/quant_cuda-0.0.0-cp310-cp310-win_amd64.whl", environment=True) url = "https://github.com/jllllll/GPTQ-for-LLaMa-Wheels/raw/main/quant_cuda-0.0.0-cp310-cp310-win_amd64.whl"
if sys.platform.startswith("linux"):
url = "https://github.com/jllllll/GPTQ-for-LLaMa-Wheels/raw/Linux-x64/quant_cuda-0.0.0-cp310-cp310-linux_x86_64.whl"
result = run_cmd("python -m pip install " + url, environment=True)
if result.returncode == 0: if result.returncode == 0:
print("Wheel installation success!") print("Wheel installation success!")
else: else: