Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-10-31 14:40:15 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix wrong pytorch version on Linux+CPU

Browse Source 
It was installing nvidia wheels
This commit is contained in:
oobabooga 2023-07-07 20:40:31 -03:00 committed by GitHub
parent 564a8c507f
commit bb79037ebd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
webui.py
View File

@ -87,8 +87,14 @@ def install_dependencies():
elif gpuchoice == "b": elif gpuchoice == "b":
print("AMD GPUs are not supported. Exiting...") print("AMD GPUs are not supported. Exiting...")
sys.exit() sys.exit()
elif gpuchoice == "c" or gpuchoice == "d": elif gpuchoice == "c":
run_cmd("conda install -y -k ninja git && python -m pip install torch torchvision torchaudio", assert_success=True, environment=True) run_cmd("conda install -y -k ninja git && python -m pip install torch torchvision torchaudio", assert_success=True, environment=True)
elif gpuchoice == "d":
if sys.platform.startswith("linux"):
run_cmd("conda install -y -k ninja git && python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu", assert_success=True, environment=True)
else:
run_cmd("conda install -y -k ninja git && python -m pip install torch torchvision torchaudio", assert_success=True, environment=True)
else: else:
print("Invalid choice. Exiting...") print("Invalid choice. Exiting...")
sys.exit() sys.exit()