updated atsetup.bat
This commit is contained in:
erew123 authored and GitHub committed
1 parent
36bfb17eda
commit
4a116e359a
1 file changed
+67
-63
+67
-63
@@ -3,6 +3,28 @@ cd /D "%~dp0"
|
||||
|
||||
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
|
||||
curl --version >nul 2>&1
|
||||
if "%ERRORLEVEL%" NEQ "0" (
|
||||
@@ -14,14 +36,13 @@ if "%ERRORLEVEL%" NEQ "0" (
|
||||
:MainMenu
|
||||
cls
|
||||
echo.
|
||||
echo Welcome to the AllTalk Setup Utility
|
||||
echo %L_BLUE%ALLTALK WINDOWS SETUP UTILITY%RESET%
|
||||
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 1) I am using AllTalk as part of Text-generation-webui
|
||||
echo 2) I am using AllTalk as a Standalone Application
|
||||
echo.
|
||||
echo 9) Exit/Quit
|
||||
echo 9) %L_RED%Exit/Quit%RESET%
|
||||
echo.
|
||||
set /p UserOption=" Enter your choice: "
|
||||
|
||||
@@ -33,20 +54,26 @@ goto MainMenu
|
||||
:WebUIMenu
|
||||
cls
|
||||
echo.
|
||||
echo You have selected Text-generation-webui.
|
||||
echo %L_BLUE%TEXT-GENERATION-WEBUI SETUP%RESET%
|
||||
echo.
|
||||
echo Please ensure you have started your Text-generation-webui Python environment. If you have NOT done this,
|
||||
echo please run "cmd_windows.bat" in the text-generation-webui folder and then re-run "atsetup.bat"
|
||||
echo Please ensure you have started your Text-generation-webui Python
|
||||
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 1) Install the requirements for a Nvidia machine.
|
||||
echo 2) Install the requirements for a AMD or MAC machine.
|
||||
echo BASE REQUIREMENTS
|
||||
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 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 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) Exit/Quit
|
||||
echo 9) %L_RED%Exit/Quit%RESET%
|
||||
echo.
|
||||
set /p WebUIOption="Enter your choice: "
|
||||
if "%WebUIOption%"=="1" goto InstallNvidiaTextGen
|
||||
@@ -62,14 +89,19 @@ goto WebUIMenu
|
||||
:StandaloneMenu
|
||||
cls
|
||||
echo.
|
||||
echo You have selected Standalone Application Setup.
|
||||
echo %L_BLUE%ALLTALK STANDALONE APPLICATION SETUP%RESET%
|
||||
echo.
|
||||
echo 1) Install AllTalk's custom Python environment (Setup AllTalk as a standalone Application)
|
||||
echo 2) Delete AllTalk's custom Python environment (You will need to run 1 again after)
|
||||
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 BASE REQUIREMENTS
|
||||
echo 1) Install AllTalk as a Standalone Application
|
||||
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.
|
||||
set /p StandaloneOption="Enter your choice: "
|
||||
if "%StandaloneOption%"=="1" goto InstallCustomStandalone
|
||||
@@ -90,11 +122,9 @@ if %ERRORLEVEL% neq 0 (
|
||||
goto WebUIMenu
|
||||
)
|
||||
Echo.
|
||||
echo *************************************************************************************
|
||||
echo.
|
||||
Echo Nvidia machine requirements installed successfully.
|
||||
Echo.
|
||||
echo *************************************************************************************
|
||||
pause
|
||||
goto WebUIMenu
|
||||
|
||||
@@ -109,11 +139,9 @@ if %ERRORLEVEL% neq 0 (
|
||||
goto WebUIMenu
|
||||
)
|
||||
Echo.
|
||||
echo *************************************************************************************
|
||||
echo.
|
||||
Echo Other machine requirements installed successfully.
|
||||
Echo.
|
||||
echo *************************************************************************************
|
||||
pause
|
||||
goto WebUIMenu
|
||||
|
||||
@@ -128,13 +156,10 @@ if %ERRORLEVEL% neq 0 (
|
||||
goto WebUIMenu
|
||||
)
|
||||
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 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
|
||||
goto WebUIMenu
|
||||
|
||||
@@ -163,11 +188,9 @@ if %ERRORLEVEL% neq 0 (
|
||||
goto WebUIMenu
|
||||
)
|
||||
Echo.
|
||||
echo *************************************************************************************
|
||||
echo.
|
||||
Echo DeepSpeed installed successfully.
|
||||
Echo.
|
||||
echo *************************************************************************************
|
||||
del deepspeed-0.11.2+cuda118-cp311-cp311-win_amd64.whl
|
||||
pause
|
||||
goto WebUIMenu
|
||||
@@ -197,11 +220,9 @@ if %ERRORLEVEL% neq 0 (
|
||||
goto WebUIMenu
|
||||
)
|
||||
Echo.
|
||||
echo *************************************************************************************
|
||||
echo.
|
||||
Echo DeepSpeed installed successfully.
|
||||
Echo.
|
||||
echo *************************************************************************************
|
||||
del deepspeed-0.11.2+cuda121-cp311-cp311-win_amd64.whl
|
||||
pause
|
||||
goto WebUIMenu
|
||||
@@ -216,11 +237,9 @@ if %ERRORLEVEL% neq 0 (
|
||||
pause
|
||||
goto WebUIMenu
|
||||
)
|
||||
echo *************************************************************************************
|
||||
echo.
|
||||
Echo DeepSpeed uninstalled successfully.
|
||||
Echo.
|
||||
echo *************************************************************************************
|
||||
pause
|
||||
goto WebUIMenu
|
||||
|
||||
@@ -229,16 +248,15 @@ Python diagnostics.py
|
||||
if %ERRORLEVEL% neq 0 (
|
||||
echo.
|
||||
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
|
||||
goto WebUIMenu
|
||||
)
|
||||
Echo.
|
||||
echo *************************************************************************************
|
||||
echo.
|
||||
Echo Diagnostics.log generated. Please scroll up to look over the log.
|
||||
Echo.
|
||||
echo *************************************************************************************
|
||||
pause
|
||||
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 )
|
||||
|
||||
@rem Ask user for the type of requirements to install
|
||||
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 1. Nvidia graphics card machines
|
||||
echo 2. Other machines (mac, amd, etc)
|
||||
echo.
|
||||
echo *************************************************************************************
|
||||
set /p UserChoice="Enter your choice (1 or 2): "
|
||||
|
||||
@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 exit >> start_alltalk.bat
|
||||
Echo.
|
||||
Echo *************************************************************************************
|
||||
Echo.
|
||||
Echo "start_alltalk.bat" has been created.
|
||||
Echo.
|
||||
Echo You can now start AllTalk with "start_alltalk.bat"
|
||||
Echo.
|
||||
Echo *************************************************************************************
|
||||
Echo start_alltalk.bat has been created.
|
||||
Echo You can now start AllTalk with %L_YELLOW%start_alltalk.bat%RESET%
|
||||
Echo.
|
||||
pause
|
||||
goto StandaloneMenu
|
||||
@@ -369,7 +380,7 @@ goto StandaloneMenu
|
||||
@rem Check if the alltalk_environment directory exists
|
||||
if not exist "%cd%\alltalk_environment\" (
|
||||
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.
|
||||
pause
|
||||
goto StandaloneMenu
|
||||
@@ -377,7 +388,7 @@ if not exist "%cd%\alltalk_environment\" (
|
||||
@rem Check if a Conda environment is active
|
||||
if not defined CONDA_PREFIX goto NoCondaEnvDeleteCustomStandalone
|
||||
@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
|
||||
:NoCondaEnvDeleteCustomStandalone
|
||||
echo Deleting "alltalk_environment". Please wait.
|
||||
@@ -385,18 +396,16 @@ rd /s /q "alltalk_environment"
|
||||
del start_alltalk.bat
|
||||
if %ERRORLEVEL% neq 0 (
|
||||
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.
|
||||
pause
|
||||
goto StandaloneMenu
|
||||
)
|
||||
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 *************************************************************************************
|
||||
pause
|
||||
goto StandaloneMenu
|
||||
|
||||
@@ -434,13 +443,10 @@ if %ERRORLEVEL% neq 0 (
|
||||
goto StandaloneMenu
|
||||
)
|
||||
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 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
|
||||
goto StandaloneMenu
|
||||
|
||||
@@ -478,11 +484,9 @@ if %ERRORLEVEL% neq 0 (
|
||||
goto StandaloneMenu
|
||||
)
|
||||
Echo.
|
||||
Echo *************************************************************************************
|
||||
Echo.
|
||||
Echo Diagnostics.log generated. Please scroll up to look over the log.
|
||||
Echo.
|
||||
Echo *************************************************************************************
|
||||
pause
|
||||
goto StandaloneMenu
|
||||
|
||||
|
||||
Reference in new issue
Block a user