From c5e0ab7174e37e26c69d16c4830d7e9b371805a8 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Fri, 22 Sep 2023 14:50:27 -0700
Subject: [PATCH] Minor bug fix

---
 one_click.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/one_click.py b/one_click.py
index 8df3e48b..d88f88fb 100644
--- a/one_click.py
+++ b/one_click.py
@@ -120,7 +120,7 @@ def install_webui():
 
     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"
-    elif is_macos() and choice == "B":
+    elif not is_macos() and choice == "B":
         if is_linux():
             install_pytorch = "python -m pip install torch==2.0.1+rocm5.4.2 torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.4.2"
         else: