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
+67 -63
Download Patch File Download Diff File Expand all files Collapse all files
atsetup.bat
+67 -63
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,14 +36,13 @@ 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 9) Exit/Quit
echo. echo.
set /p UserOption=" Enter your choice: " set /p UserOption=" Enter your choice: "
@@ -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 2) Install the requirements for a %L_GREEN%AMD or MAC machine%RESET%.
echo.
echo OPTIONAL
echo 3) Install the Finetuning requirements. echo 3) Install the Finetuning requirements.
echo 4) Install "DeepSpeed v11.2 for CUDA 11.8 & Python 3.11.x". 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 12.1 & 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 6) Uninstall DeepSpeed.
echo.
echo. OTHER
echo 7) Generate a diagnostics file. echo 7) Generate a diagnostics file.
echo. echo.
echo 9) Exit/Quit 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
@@ -90,11 +122,9 @@ if %ERRORLEVEL% neq 0 (
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
@@ -109,11 +139,9 @@ if %ERRORLEVEL% neq 0 (
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
@@ -128,13 +156,10 @@ if %ERRORLEVEL% neq 0 (
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
@@ -163,11 +188,9 @@ if %ERRORLEVEL% neq 0 (
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
@@ -197,11 +220,9 @@ if %ERRORLEVEL% neq 0 (
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
@@ -216,11 +237,9 @@ if %ERRORLEVEL% neq 0 (
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
@@ -229,16 +248,15 @@ 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
@@ -434,13 +443,10 @@ if %ERRORLEVEL% neq 0 (
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
@@ -478,11 +484,9 @@ if %ERRORLEVEL% neq 0 (
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