Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-25 09:19:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix pytorch installation on Linux

Browse Source
This commit is contained in:
oobabooga 2023-09-22 14:34:30 -07:00
parent b1467bd064
commit a0c7d764b5
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
one_click.py
View File

@ -118,7 +118,7 @@ def install_webui():
install_git = "conda install -y -k ninja git" install_git = "conda install -y -k ninja git"
install_pytorch = "python -m pip install torch torchvision torchaudio" install_pytorch = "python -m pip install torch torchvision torchaudio"
if choice == "A": if is_windows() and choice == "A":
install_pytorch = "python -m pip install torch==2.0.1+cu117 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117" install_pytorch = "python -m pip install torch==2.0.1+cu117 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117"
elif is_macos() and choice == "B": elif is_macos() and choice == "B":
if is_linux(): if is_linux():