From 58797a9eb5f386cc8262f6d8f1a152494249c28d Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Sun, 17 Aug 2025 14:17:20 -0700
Subject: [PATCH] Minor change after 9651b5c873649e5c967142f9f78e7ad6cf59aaf5

---
 one_click.py | 12 ++----------
 1 file changed, 2 insertions(+), 10 deletions(-)

diff --git a/one_click.py b/one_click.py
index 486e893e..67be9e4b 100644
--- a/one_click.py
+++ b/one_click.py
@@ -135,9 +135,7 @@ def get_pytorch_install_command(gpu_choice):
     """Get PyTorch installation command based on GPU choice"""
     base_cmd = f"python -m pip install torch=={TORCH_VERSION} "
 
-    if gpu_choice == "NVIDIA":
-        return base_cmd + "--index-url https://download.pytorch.org/whl/cu124"
-    elif gpu_choice == "NVIDIA_CUDA128":
+    if gpu_choice == "NVIDIA_CUDA128":
         return "python -m pip install torch==2.7.1 --index-url https://download.pytorch.org/whl/cu128"
     elif gpu_choice == "AMD":
         return base_cmd + "--index-url https://download.pytorch.org/whl/rocm6.2.4"
@@ -156,9 +154,7 @@ def get_pytorch_update_command(gpu_choice):
     """Get PyTorch update command based on GPU choice"""
     base_cmd = f"python -m pip install --upgrade torch=={TORCH_VERSION} "
 
-    if gpu_choice == "NVIDIA":
-        return f"{base_cmd} --index-url https://download.pytorch.org/whl/cu124"
-    elif gpu_choice == "NVIDIA_CUDA128":
+    if gpu_choice == "NVIDIA_CUDA128":
         return "python -m pip install --upgrade torch==2.7.1 --index-url https://download.pytorch.org/whl/cu128"
     elif gpu_choice == "AMD":
         return f"{base_cmd} --index-url https://download.pytorch.org/whl/rocm6.2.4"
@@ -181,8 +177,6 @@ def get_requirements_file(gpu_choice):
         file_name = f"requirements_apple_{'intel' if is_x86_64() else 'silicon'}.txt"
     elif gpu_choice in ["INTEL", "NONE"]:
         file_name = f"requirements_cpu_only{'_noavx2' if not cpu_has_avx2() else ''}.txt"
-    elif gpu_choice == "NVIDIA":
-        file_name = f"requirements{'_noavx2' if not cpu_has_avx2() else ''}.txt"
     elif gpu_choice == "NVIDIA_CUDA128":
         file_name = f"requirements_cuda128{'_noavx2' if not cpu_has_avx2() else ''}.txt"
     else:
@@ -330,8 +324,6 @@ def install_webui():
                 cmd_flags_file.write("\n--cpu\n")
 
     # Handle CUDA version display
-    elif any((is_windows(), is_linux())) and gpu_choice == "NVIDIA":
-        print("CUDA: 12.4")
     elif any((is_windows(), is_linux())) and gpu_choice == "NVIDIA_CUDA128":
         print("CUDA: 12.8")