From c8ce2e777b2360d8ab5b245efeebf801358cf200 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Thu, 25 May 2023 10:57:52 -0300
Subject: [PATCH] Add instructions for CPU mode users

---
 webui.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/webui.py b/webui.py
index faf0452d..70d473ac 100644
--- a/webui.py
+++ b/webui.py
@@ -58,6 +58,9 @@ def install_dependencies():
     print()
     gpuchoice = input("Input> ").lower()
 
+    if gpuchoice == "d":
+        print("\nOnce the installation ends, make sure to open webui.py with a text editor and add the --cpu flag to CMD_FLAGS.\n")
+
     # Install the version of PyTorch needed
     if gpuchoice == "a":
         run_cmd("conda install -y -k pytorch[version=2,build=py3.10_cuda11.7*] torchvision torchaudio pytorch-cuda=11.7 cuda-toolkit ninja git -c pytorch -c nvidia/label/cuda-11.7.0 -c nvidia", assert_success=True, environment=True)