diff --git a/app.py b/app.py index 93cfb3dd..c988fe50 100644 --- a/app.py +++ b/app.py @@ -856,7 +856,7 @@ class LoLLMsWebUI(LoLLMsAPPI): return jsonify({'status':False, "error": "Couldn't install cuda toolkit. Make sure you are running from conda environment"}) ASCIIColors.green("Cuda toolkit installed successfully") ASCIIColors.yellow("Removing pytorch") - res = subprocess.check_call(["pip","uninstall","torch", "torchvision", "torchaudio"]) + res = subprocess.check_call(["pip","uninstall","torch", "torchvision", "torchaudio", "-y"]) ASCIIColors.green("PyTorch unstalled successfully") ASCIIColors.yellow("Installing pytorch with cuda support") res = subprocess.check_call(["pip","install","--upgrade","torch==2.0.1+cu117", "torchvision", "torchaudio", "--index-url", "https://download.pytorch.org/whl/cu117","no-cache"])