AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-09-20 00:07:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make N the "None" install option

Browse Source
This commit is contained in:
oobabooga 2023-09-23 05:25:06 -07:00
parent 0306b61bb0
commit 639723845a
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
one_click.py
View File

@ -109,15 +109,15 @@ def install_webui():
print("B) AMD (Linux/MacOS only. Requires ROCm SDK 5.4.2/5.4.3 on Linux)")
print("C) Apple M Series")
print("D) Intel Arc (IPEX)")
print("E) None (I want to run models in CPU mode)")
print("N) None (I want to run models in CPU mode)")
print()
choice = input("Input> ").upper()
while choice not in 'ABCDE':
while choice not in 'ABCDN':
print("Invalid choice. Please try again.")
choice = input("Input> ").upper()
if choice == "E":
if choice == "N":
print_big_message("Once the installation ends, make sure to open CMD_FLAGS.txt with\na text editor and add the --cpu flag.")
# Find the proper Pytorch installation command
@ -132,7 +132,7 @@ def install_webui():
else:
print("AMD GPUs are only supported on Linux. Exiting...")
sys.exit()
elif is_linux() and (choice == "C" or choice == "E"):
elif is_linux() and (choice == "C" or choice == "N"):
install_pytorch = "python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu"
elif choice == "D":
install_pytorch = "python -m pip install torch==2.0.1a0 torchvision==0.15.2a0 intel_extension_for_pytorch==2.0.110+xpu -f https://developer.intel.com/ipex-whl-stable-xpu"