From 86648d4085bec7b0aa3ebc0e1e5ced9b4f0c1782 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Fri, 22 Sep 2023 08:13:11 -0700
Subject: [PATCH] Remove CUDA, keep only pytorch

---
 CMD_FLAGS.txt | 1 +
 webui.py      | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/CMD_FLAGS.txt b/CMD_FLAGS.txt
index e69de29b..47fe8405 100644
--- a/CMD_FLAGS.txt
+++ b/CMD_FLAGS.txt
@@ -0,0 +1 @@
+--listen
diff --git a/webui.py b/webui.py
index aaa77885..4225cd7e 100644
--- a/webui.py
+++ b/webui.py
@@ -122,7 +122,7 @@ def install_dependencies():
 
     # Install Pytorch
     if gpuchoice == "a":
-        run_cmd('conda install -y -k cuda ninja git -c nvidia/label/cuda-11.7.0 -c nvidia && python -m pip install torch==2.0.1+cu117 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117', assert_success=True, environment=True)
+        run_cmd('conda install -y -k ninja git && python -m pip install torch==2.0.1+cu117 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117', assert_success=True, environment=True)
     elif gpuchoice == "b" and not is_macos():
         if is_linux():
             run_cmd('conda install -y -k ninja git && python -m pip install torch==2.0.1+rocm5.4.2 torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.4.2', assert_success=True, environment=True)