kromskii2/alltalk_tts
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

updated atsetup.bat

Browse Source
This commit is contained in:
erew123 authored and GitHub committed 2024-01-12 12:21:07 +00:00
1 parent 36bfb17eda
commit 4a116e359a
1 file changed
+116 -112
Download Patch File Download Diff File Expand all files Collapse all files
atsetup.bat
+116 -112
View File
@@ -3,6 +3,28 @@ cd /D "%~dp0"
set PATH=%PATH%;%SystemRoot%\system32 set PATH=%PATH%;%SystemRoot%\system32
:: Generate the ESC character
for /F %%a in ('echo prompt $E ^| cmd') do set "ESC=%%a"
:: Standard Colors
set "BLACK=%ESC%[30m"
set "RED=%ESC%[31m"
set "GREEN=%ESC%[32m"
set "YELLOW=%ESC%[33m"
set "BLUE=%ESC%[34m"
set "MAGENTA=%ESC%[35m"
set "CYAN=%ESC%[36m"
set "WHITE=%ESC%[37m"
set "L_BLACK=%ESC%[90m"
set "L_RED=%ESC%[91m"
set "L_GREEN=%ESC%[92m"
set "L_YELLOW=%ESC%[93m"
set "L_BLUE=%ESC%[94m"
set "L_MAGENTA=%ESC%[95m"
set "L_CYAN=%ESC%[96m"
set "L_WHITE=%ESC%[97m"
set "RESET=%ESC%[0m"
@rem Check if curl is available @rem Check if curl is available
curl --version >nul 2>&1 curl --version >nul 2>&1
if "%ERRORLEVEL%" NEQ "0" ( if "%ERRORLEVEL%" NEQ "0" (
@@ -14,16 +36,15 @@ if "%ERRORLEVEL%" NEQ "0" (
:MainMenu :MainMenu
cls cls
echo. echo.
echo Welcome to the AllTalk Setup Utility echo %L_BLUE%ALLTALK WINDOWS SETUP UTILITY%RESET%
echo. echo.
echo Please select an option: echo INSTALLATION TYPE
echo 1) I am using AllTalk as part of %L_GREEN%Text-generation-webui%RESET%
echo 2) I am using AllTalk as a %L_GREEN%Standalone Application%RESET%
echo. echo.
echo 1) I am using AllTalk as part of Text-generation-webui echo 9) %L_RED%Exit/Quit%RESET%
echo 2) I am using AllTalk as a Standalone Application
echo. echo.
echo 9) Exit/Quit set /p UserOption=" Enter your choice: "
echo.
set /p UserOption="Enter your choice: "
if "%UserOption%"=="1" goto WebUIMenu if "%UserOption%"=="1" goto WebUIMenu
if "%UserOption%"=="2" goto StandaloneMenu if "%UserOption%"=="2" goto StandaloneMenu
@@ -33,20 +54,26 @@ goto MainMenu
:WebUIMenu :WebUIMenu
cls cls
echo. echo.
echo You have selected Text-generation-webui. echo %L_BLUE%TEXT-GENERATION-WEBUI SETUP%RESET%
echo. echo.
echo Please ensure you have started your Text-generation-webui Python environment. If you have NOT done this, echo Please ensure you have started your Text-generation-webui Python
echo please run "cmd_windows.bat" in the text-generation-webui folder and then re-run "atsetup.bat" echo environment. If you have NOT done this, please run %L_GREEN%cmd_windows.bat%RESET%
echo in the %L_GREEN%text-generation-webui%RESET% folder and then re-run this script.
echo. echo.
echo 1) Install the requirements for a Nvidia machine. echo BASE REQUIREMENTS
echo 2) Install the requirements for a AMD or MAC machine. echo 1) Install the requirements for a %L_GREEN%Nvidia machine%RESET%.
echo 3) Install the Finetuning requirements. echo 2) Install the requirements for a %L_GREEN%AMD or MAC machine%RESET%.
echo 4) Install "DeepSpeed v11.2 for CUDA 11.8 & Python 3.11.x".
echo 5) Install "DeepSpeed v11.2 for CUDA 12.1 & Python 3.11.x".
echo 6) Uninstall DeepSpeed.
echo 7) Generate a diagnostics file.
echo. echo.
echo 9) Exit/Quit echo OPTIONAL
echo 3) Install the Finetuning requirements.
echo 4) Install DeepSpeed v11.2 for CUDA %L_GREEN%11.8%RESET% and Python-3.11.x.
echo 5) Install DeepSpeed v11.2 for CUDA %L_GREEN%12.1%RESET% and Python-3.11.x.
echo 6) Uninstall DeepSpeed.
echo.
echo. OTHER
echo 7) Generate a diagnostics file.
echo.
echo 9) %L_RED%Exit/Quit%RESET%
echo. echo.
set /p WebUIOption="Enter your choice: " set /p WebUIOption="Enter your choice: "
if "%WebUIOption%"=="1" goto InstallNvidiaTextGen if "%WebUIOption%"=="1" goto InstallNvidiaTextGen
@@ -62,14 +89,19 @@ goto WebUIMenu
:StandaloneMenu :StandaloneMenu
cls cls
echo. echo.
echo You have selected Standalone Application Setup. echo %L_BLUE%ALLTALK STANDALONE APPLICATION SETUP%RESET%
echo. echo.
echo 1) Install AllTalk's custom Python environment (Setup AllTalk as a standalone Application) echo BASE REQUIREMENTS
echo 2) Delete AllTalk's custom Python environment (You will need to run 1 again after) echo 1) Install AllTalk as a Standalone Application
echo 3) Install the Finetuning requirements (You need to have created the AllTalk custom environment)
echo 4) Generate a diagnostics file (You need to have created the AllTalk custom environment)
echo. echo.
echo 9) Exit/Quit echo OPTIONAL
echo 2) Delete AllTalk's custom Python environment
echo 3) Install the Finetuning requirements
echo.
echo. OTHER
echo 4) Generate a diagnostics file
echo.
echo 9) %L_RED%Exit/Quit%RESET%
echo. echo.
set /p StandaloneOption="Enter your choice: " set /p StandaloneOption="Enter your choice: "
if "%StandaloneOption%"=="1" goto InstallCustomStandalone if "%StandaloneOption%"=="1" goto InstallCustomStandalone
@@ -83,18 +115,16 @@ goto StandaloneMenu
pip install -r requirements_nvidia.txt pip install -r requirements_nvidia.txt
if %ERRORLEVEL% neq 0 ( if %ERRORLEVEL% neq 0 (
echo. echo.
echo There was an error installing the Nvidia requirements. echo There was an error installing the Nvidia requirements.
echo Press any key to return to the menu. echo Press any key to return to the menu.
echo. echo.
pause pause
goto WebUIMenu goto WebUIMenu
) )
Echo. Echo.
echo *************************************************************************************
echo. echo.
Echo Nvidia machine requirements installed successfully. Echo Nvidia machine requirements installed successfully.
Echo. Echo.
echo *************************************************************************************
pause pause
goto WebUIMenu goto WebUIMenu
@@ -102,18 +132,16 @@ goto WebUIMenu
pip install -r requirements_other.txt pip install -r requirements_other.txt
if %ERRORLEVEL% neq 0 ( if %ERRORLEVEL% neq 0 (
echo. echo.
echo There was an error installing the Other requirements. echo There was an error installing the Other requirements.
echo Press any key to return to the menu. echo Press any key to return to the menu.
echo. echo.
pause pause
goto WebUIMenu goto WebUIMenu
) )
Echo. Echo.
echo *************************************************************************************
echo. echo.
Echo Other machine requirements installed successfully. Echo Other machine requirements installed successfully.
Echo. Echo.
echo *************************************************************************************
pause pause
goto WebUIMenu goto WebUIMenu
@@ -121,20 +149,17 @@ goto WebUIMenu
pip install -r requirements_finetune.txt pip install -r requirements_finetune.txt
if %ERRORLEVEL% neq 0 ( if %ERRORLEVEL% neq 0 (
echo. echo.
echo There was an error installing the Finetune requirements. echo There was an error installing the Finetune requirements.
echo Press any key to return to the menu. echo Press any key to return to the menu.
echo. echo.
pause pause
goto WebUIMenu goto WebUIMenu
) )
Echo. Echo.
Echo ****************************************************************************************************** Echo Finetune requirements installed successfully. Please ensure you have installed CUDA 11.8.
Echo Finetune needs access to %L_GREEN%cublas64_11.dll%RESET% and will error if this is not setup correctly.
Echo Please see here for details %L_GREEN%https://github.com/erew123/alltalk_tts#-finetuning-a-model%RESET%
Echo. Echo.
Echo Finetune requirements installed successfully. Please ensure you have installed CUDA 11.8 requirements.
Echo Finetune needs access to cublas64_11.dll and will error if this is not setup correctly.
Echo Please see here for details https://github.com/erew123/alltalk_tts#-finetuning-a-model
Echo.
Echo ******************************************************************************************************
pause pause
goto WebUIMenu goto WebUIMenu
@@ -143,9 +168,9 @@ echo Downloading DeepSpeed...
curl -LO https://github.com/erew123/alltalk_tts/releases/download/deepspeed/deepspeed-0.11.2+cuda118-cp311-cp311-win_amd64.whl curl -LO https://github.com/erew123/alltalk_tts/releases/download/deepspeed/deepspeed-0.11.2+cuda118-cp311-cp311-win_amd64.whl
if %ERRORLEVEL% neq 0 ( if %ERRORLEVEL% neq 0 (
echo. echo.
echo Failed to download DeepSpeed wheel file. echo Failed to download DeepSpeed wheel file.
echo Please check your internet connection or try again later. echo Please check your internet connection or try again later.
echo Press any key to return to the menu. echo Press any key to return to the menu.
echo. echo.
pause pause
goto WebUIMenu goto WebUIMenu
@@ -155,19 +180,17 @@ echo Installing DeepSpeed...
pip install deepspeed-0.11.2+cuda118-cp311-cp311-win_amd64.whl pip install deepspeed-0.11.2+cuda118-cp311-cp311-win_amd64.whl
if %ERRORLEVEL% neq 0 ( if %ERRORLEVEL% neq 0 (
echo. echo.
echo Failed to install DeepSpeed. echo Failed to install DeepSpeed.
echo Please check if the wheel file is compatible with your system. echo Please check if the wheel file is compatible with your system.
echo Press any key to return to the menu. echo Press any key to return to the menu.
echo. echo.
pause pause
goto WebUIMenu goto WebUIMenu
) )
Echo. Echo.
echo *************************************************************************************
echo. echo.
Echo DeepSpeed installed successfully. Echo DeepSpeed installed successfully.
Echo. Echo.
echo *************************************************************************************
del deepspeed-0.11.2+cuda118-cp311-cp311-win_amd64.whl del deepspeed-0.11.2+cuda118-cp311-cp311-win_amd64.whl
pause pause
goto WebUIMenu goto WebUIMenu
@@ -177,31 +200,29 @@ echo Downloading DeepSpeed...
curl -LO https://github.com/erew123/alltalk_tts/releases/download/deepspeed/deepspeed-0.11.2+cuda121-cp311-cp311-win_amd64.whl curl -LO https://github.com/erew123/alltalk_tts/releases/download/deepspeed/deepspeed-0.11.2+cuda121-cp311-cp311-win_amd64.whl
if %ERRORLEVEL% neq 0 ( if %ERRORLEVEL% neq 0 (
echo. echo.
echo Failed to download DeepSpeed wheel file. echo Failed to download DeepSpeed wheel file.
echo Please check your internet connection or try again later. echo Please check your internet connection or try again later.
echo Press any key to return to the menu. echo Press any key to return to the menu.
echo. echo.
pause pause
goto WebUIMenu goto WebUIMenu
) )
echo DeepSpeed wheel file downloaded successfully. echo DeepSpeed wheel file downloaded successfully.
echo Installing DeepSpeed... echo Installing DeepSpeed...
pip install deepspeed-0.11.2+cuda121-cp311-cp311-win_amd64.whl pip install deepspeed-0.11.2+cuda121-cp311-cp311-win_amd64.whl
if %ERRORLEVEL% neq 0 ( if %ERRORLEVEL% neq 0 (
echo. echo.
echo Failed to install DeepSpeed. echo Failed to install DeepSpeed.
echo Please check if the wheel file is compatible with your system. echo Please check if the wheel file is compatible with your system.
echo Press any key to return to the menu. echo Press any key to return to the menu.
echo. echo.
pause pause
goto WebUIMenu goto WebUIMenu
) )
Echo. Echo.
echo *************************************************************************************
echo. echo.
Echo DeepSpeed installed successfully. Echo DeepSpeed installed successfully.
Echo. Echo.
echo *************************************************************************************
del deepspeed-0.11.2+cuda121-cp311-cp311-win_amd64.whl del deepspeed-0.11.2+cuda121-cp311-cp311-win_amd64.whl
pause pause
goto WebUIMenu goto WebUIMenu
@@ -210,17 +231,15 @@ goto WebUIMenu
pip uninstall deepspeed pip uninstall deepspeed
if %ERRORLEVEL% neq 0 ( if %ERRORLEVEL% neq 0 (
echo. echo.
echo There was an error uninstalling DeepSpeed. echo There was an error uninstalling DeepSpeed.
echo Press any key to return to the menu. echo Press any key to return to the menu.
echo. echo.
pause pause
goto WebUIMenu goto WebUIMenu
) )
echo *************************************************************************************
echo. echo.
Echo DeepSpeed uninstalled successfully. Echo DeepSpeed uninstalled successfully.
Echo. Echo.
echo *************************************************************************************
pause pause
goto WebUIMenu goto WebUIMenu
@@ -228,17 +247,16 @@ goto WebUIMenu
Python diagnostics.py Python diagnostics.py
if %ERRORLEVEL% neq 0 ( if %ERRORLEVEL% neq 0 (
echo. echo.
echo There was an error running diagnostics. Have you correctly started your echo There was an error running diagnostics. Have you correctly started your
echo Text-generation-webui Python environment with "cmd_windows.bat"? echo Text-generation-webui Python environment with %L_GREEN%cmd_windows.bat%RESET%?
echo.
pause pause
goto WebUIMenu goto WebUIMenu
) )
Echo. Echo.
echo *************************************************************************************
echo. echo.
Echo Diagnostics.log generated. Please scroll up to look over the log. Echo Diagnostics.log generated. Please scroll up to look over the log.
Echo. Echo.
echo *************************************************************************************
pause pause
goto WebUIMenu goto WebUIMenu
@@ -303,14 +321,12 @@ if not exist "%INSTALL_ENV_DIR%\python.exe" ( echo. && echo Conda environment is
call "%CONDA_ROOT_PREFIX%\condabin\conda.bat" activate "%INSTALL_ENV_DIR%" || ( echo. && echo Miniconda hook not found. && goto end ) call "%CONDA_ROOT_PREFIX%\condabin\conda.bat" activate "%INSTALL_ENV_DIR%" || ( echo. && echo Miniconda hook not found. && goto end )
@rem Ask user for the type of requirements to install @rem Ask user for the type of requirements to install
echo *************************************************************************************
echo. echo.
echo Choose the type of requirements to install (this will take 10-20 minutes to install): echo %L_BLUE%Choose the type of requirements to install%RESET%
echo. echo.
echo 1. Nvidia graphics card machines echo 1. Nvidia graphics card machines
echo 2. Other machines (mac, amd, etc) echo 2. Other machines (mac, amd, etc)
echo. echo.
echo *************************************************************************************
set /p UserChoice="Enter your choice (1 or 2): " set /p UserChoice="Enter your choice (1 or 2): "
@rem Install requirements based on user choice @rem Install requirements based on user choice
@@ -354,13 +370,8 @@ echo call "%CONDA_ROOT_PREFIX%\condabin\conda.bat" activate "%INSTALL_ENV_DIR%"
echo call python script.py >> start_alltalk.bat echo call python script.py >> start_alltalk.bat
echo exit >> start_alltalk.bat echo exit >> start_alltalk.bat
Echo. Echo.
Echo ************************************************************************************* Echo start_alltalk.bat has been created.
Echo. Echo You can now start AllTalk with %L_YELLOW%start_alltalk.bat%RESET%
Echo "start_alltalk.bat" has been created.
Echo.
Echo You can now start AllTalk with "start_alltalk.bat"
Echo.
Echo *************************************************************************************
Echo. Echo.
pause pause
goto StandaloneMenu goto StandaloneMenu
@@ -369,7 +380,7 @@ goto StandaloneMenu
@rem Check if the alltalk_environment directory exists @rem Check if the alltalk_environment directory exists
if not exist "%cd%\alltalk_environment\" ( if not exist "%cd%\alltalk_environment\" (
echo. echo.
echo "alltalk_environment" directory does not exist. No need to delete. echo %L_GREEN%alltalk_environment%RESET% directory does not exist. No need to delete.
echo. echo.
pause pause
goto StandaloneMenu goto StandaloneMenu
@@ -377,7 +388,7 @@ if not exist "%cd%\alltalk_environment\" (
@rem Check if a Conda environment is active @rem Check if a Conda environment is active
if not defined CONDA_PREFIX goto NoCondaEnvDeleteCustomStandalone if not defined CONDA_PREFIX goto NoCondaEnvDeleteCustomStandalone
@rem Deactivate Conda environment if it's active @rem Deactivate Conda environment if it's active
Echo Exiting the Conda Environment. Please run "atsetup.bat" again and delete the environment. Echo Exiting the Conda Environment. Please run %L_GREEN%atsetup.bat%RESET% again and delete the environment.
conda deactivate conda deactivate
:NoCondaEnvDeleteCustomStandalone :NoCondaEnvDeleteCustomStandalone
echo Deleting "alltalk_environment". Please wait. echo Deleting "alltalk_environment". Please wait.
@@ -385,18 +396,16 @@ rd /s /q "alltalk_environment"
del start_alltalk.bat del start_alltalk.bat
if %ERRORLEVEL% neq 0 ( if %ERRORLEVEL% neq 0 (
echo. echo.
echo Failed to delete "alltalk_environment" folder. echo Failed to delete alltalk_environment folder.
echo Please make sure it is not in use and try again. echo Please make sure it is not in use and try again.
echo. echo.
pause pause
goto StandaloneMenu goto StandaloneMenu
) )
echo. echo.
Echo *************************************************************************************
Echo. Echo.
echo Environment "alltalk_environment" deleted. Please set up the environment again. echo Environment %L_GREEN%alltalk_environment%RESET% deleted. Please set up the environment again.
Echo. Echo.
Echo *************************************************************************************
pause pause
goto StandaloneMenu goto StandaloneMenu
@@ -407,8 +416,8 @@ set INSTALL_ENV_DIR=%cd%\alltalk_environment\env
@rem Check if the Conda environment exists @rem Check if the Conda environment exists
if not exist "%INSTALL_ENV_DIR%\python.exe" ( if not exist "%INSTALL_ENV_DIR%\python.exe" (
echo. echo.
echo The Conda environment at "%INSTALL_ENV_DIR%" does not exist. echo The Conda environment at "%INSTALL_ENV_DIR%" does not exist.
echo Please install the environment before proceeding. echo Please install the environment before proceeding.
echo. echo.
pause pause
goto StandaloneMenu goto StandaloneMenu
@@ -417,8 +426,8 @@ if not exist "%INSTALL_ENV_DIR%\python.exe" (
call "%CONDA_ROOT_PREFIX%\condabin\conda.bat" activate "%INSTALL_ENV_DIR%" call "%CONDA_ROOT_PREFIX%\condabin\conda.bat" activate "%INSTALL_ENV_DIR%"
if errorlevel 1 ( if errorlevel 1 (
echo. echo.
echo Failed to activate the Conda environment. echo Failed to activate the Conda environment.
echo Please check your installation and try again. echo Please check your installation and try again.
echo. echo.
pause pause
goto StandaloneMenu goto StandaloneMenu
@@ -427,20 +436,17 @@ if errorlevel 1 (
pip install -r requirements_finetune.txt pip install -r requirements_finetune.txt
if %ERRORLEVEL% neq 0 ( if %ERRORLEVEL% neq 0 (
echo. echo.
echo There was an error installing the Finetune requirements. echo There was an error installing the Finetune requirements.
echo Press any key to return to the menu. echo Press any key to return to the menu.
echo. echo.
pause pause
goto StandaloneMenu goto StandaloneMenu
) )
Echo. Echo.
Echo ****************************************************************************************************** Echo Finetune requirements installed successfully. Please ensure you have installed CUDA 11.8.
Echo Finetune needs access to %L_GREEN%cublas64_11.dll%RESET% and will error if this is not setup correctly.
Echo Please see here for details %L_GREEN%https://github.com/erew123/alltalk_tts#-finetuning-a-model%RESET%
Echo. Echo.
Echo Finetune requirements installed successfully. Please ensure you have installed CUDA 11.8 requirements.
Echo Finetune needs access to cublas64_11.dll and will error if this is not setup correctly.
Echo Please see here for details https://github.com/erew123/alltalk_tts#-finetuning-a-model
Echo.
Echo ******************************************************************************************************
pause pause
goto StandaloneMenu goto StandaloneMenu
@@ -451,8 +457,8 @@ set INSTALL_ENV_DIR=%cd%\alltalk_environment\env
@rem Check if the Conda environment exists @rem Check if the Conda environment exists
if not exist "%INSTALL_ENV_DIR%\python.exe" ( if not exist "%INSTALL_ENV_DIR%\python.exe" (
echo. echo.
echo The Conda environment at "%INSTALL_ENV_DIR%" does not exist. echo The Conda environment at "%INSTALL_ENV_DIR%" does not exist.
echo Please install the environment before proceeding. echo Please install the environment before proceeding.
echo. echo.
pause pause
goto StandaloneMenu goto StandaloneMenu
@@ -461,8 +467,8 @@ if not exist "%INSTALL_ENV_DIR%\python.exe" (
call "%CONDA_ROOT_PREFIX%\condabin\conda.bat" activate "%INSTALL_ENV_DIR%" call "%CONDA_ROOT_PREFIX%\condabin\conda.bat" activate "%INSTALL_ENV_DIR%"
if errorlevel 1 ( if errorlevel 1 (
echo. echo.
echo Failed to activate the Conda environment. echo Failed to activate the Conda environment.
echo Please check your installation and try again. echo Please check your installation and try again.
echo. echo.
pause pause
goto StandaloneMenu goto StandaloneMenu
@@ -471,18 +477,16 @@ if errorlevel 1 (
Python diagnostics.py Python diagnostics.py
if %ERRORLEVEL% neq 0 ( if %ERRORLEVEL% neq 0 (
echo. echo.
echo There was an error running diagnostics. echo There was an error running diagnostics.
echo Press any key to return to the menu. echo Press any key to return to the menu.
echo. echo.
pause pause
goto StandaloneMenu goto StandaloneMenu
) )
Echo. Echo.
Echo *************************************************************************************
Echo. Echo.
Echo Diagnostics.log generated. Please scroll up to look over the log. Echo Diagnostics.log generated. Please scroll up to look over the log.
Echo. Echo.
Echo *************************************************************************************
pause pause
goto StandaloneMenu goto StandaloneMenu