From b76ff028f81a753a11ef44914794c35ffe0f589d Mon Sep 17 00:00:00 2001
From: erew123 <35898566+erew123@users.noreply.github.com>
Date: Mon, 1 Apr 2024 03:05:27 +0100
Subject: [PATCH] Fixed DeepSpeed again (sorry)

Missed a digit off the Pytorch download and fluffed the version of DeepSpeed.
---
 atsetup.bat | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/atsetup.bat b/atsetup.bat
index 84bc9e4..618bd29 100644
--- a/atsetup.bat
+++ b/atsetup.bat
@@ -387,10 +387,10 @@ pip install torch>=2.2.1+cu121 torchaudio>=2.2.1+cu121 --upgrade --force-reinsta
 echo Installing other requirements.
 echo.
 pip install -r system\requirements\requirements_standalone.txt
-curl -LO https://github.com/erew123/alltalk_tts/releases/download/DeepSpeed-14.0/deepspeed-0.14.0+cu118-cp311-cp311-win_amd64.whl
+curl -LO https://github.com/erew123/alltalk_tts/releases/download/DeepSpeed-14.0/deepspeed-0.14.0+ce78a63-cp311-cp311-win_amd64.whl
 echo Installing DeepSpeed...
-pip install deepspeed-0.14.0+cu118-cp311-cp311-win_amd64.whl
-del deepspeed-0.14.0+cu118-cp311-cp311-win_amd64.whl
+pip install deepspeed-0.14.0+ce78a63-cp311-cp311-win_amd64.whl
+del deepspeed-0.14.0+ce78a63-cp311-cp311-win_amd64.whl
 
 
 @rem Create start_environment.bat to run AllTalk environment
@@ -524,14 +524,14 @@ echo     Downloading and installing PyTorch. This step can take a long time
 echo     depending on your internet connection and hard drive speed. Please
 echo     be patient.
 echo.
-pip install torch>=2.2.1+cu121 torchaudio>=2.2.1+cu121 --upgrade --force-reinstall --extra-index-url https://download.pytorch.org/whl/cu1
+pip install torch>=2.2.1+cu121 torchaudio>=2.2.1+cu121 --upgrade --force-reinstall --extra-index-url https://download.pytorch.org/whl/cu121
 echo Installing other requirements.
 echo.
 pip install -r system\requirements\requirements_standalone.txt
-curl -LO https://github.com/erew123/alltalk_tts/releases/download/DeepSpeed-14.0/deepspeed-0.14.0+cu118-cp311-cp311-win_amd64.whl
+curl -LO https://github.com/erew123/alltalk_tts/releases/download/DeepSpeed-14.0/deepspeed-0.14.0+ce78a63-cp311-cp311-win_amd64.whl
 echo Installing DeepSpeed...
-pip install deepspeed-0.14.0+cu118-cp311-cp311-win_amd64.whl
-del deepspeed-0.14.0+cu118-cp311-cp311-win_amd64.whl
+pip install deepspeed-0.14.0+ce78a63-cp311-cp311-win_amd64.whl
+del deepspeed-0.14.0+ce78a63-cp311-cp311-win_amd64.whl
 if %ERRORLEVEL% neq 0 (
     echo.
     echo    There was an error.