From 639723845a7bf2d01a0dcd9cdce40a46177df05f Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Sat, 23 Sep 2023 05:25:06 -0700
Subject: [PATCH] Make N the "None" install option

---
 one_click.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/one_click.py b/one_click.py
index ed8c5bab..9b1b76ab 100644
--- a/one_click.py
+++ b/one_click.py
@@ -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"