682 lines
23 KiB
Batchfile
682 lines
23 KiB
Batchfile
@echo off
|
|
cd /D "%~dp0"
|
|
setlocal enabledelayedexpansion
|
|
@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"
|
|
|
|
echo "%CD%"| findstr /C:" " >nul && echo. && echo %L_BLUE%ALLTALK WINDOWS SETUP UTILITY%RESET% && echo. && echo. && echo You are trying to install AllTalk in a folder that has a space in the folder path e.g. && echo. && echo C:\%L_RED%program files%RESET%\alltalk_tts && echo. && echo This causes errors with Conda and Python scripts. Please follow this link for reference: && echo. && echo %L_CYAN%https://docs.anaconda.com/free/working-with-conda/reference/faq/#installing-anaconda%RESET% && echo. && echo Please use a folder path that has no spaces in it e.g. && echo. && echo C:\myfiles\alltalk_tts\ && echo. && echo. && pause && goto :end
|
|
|
|
@rem Check for special characters in installation path
|
|
set "SPCHARMESSAGE="WARNING: Special characters were detected in the installation path!" " This can cause the installation to fail!""
|
|
echo "%CD%"| findstr /R /C:"[!#\$%&()\*+,;<=>?@\[\]\^`{|}~]" >nul && (
|
|
call :PrintBigMessage %SPCHARMESSAGE%
|
|
)
|
|
set SPCHARMESSAGE=
|
|
|
|
@rem Check if curl is available
|
|
curl --version >nul 2>&1
|
|
if "%ERRORLEVEL%" NEQ "0" (
|
|
echo curl is not available on this system. Please install curl then re-run the script https://curl.se/
|
|
echo or perform a manual installation of a Conda Python environment.
|
|
goto end
|
|
)
|
|
|
|
:: Check for "-silent" install command-line argument
|
|
if "%1"=="-silent" goto InstallCustomStandalone
|
|
|
|
:MainMenu
|
|
cls
|
|
echo.
|
|
echo %L_BLUE%ALLTALK WINDOWS SETUP UTILITY%RESET%
|
|
echo.
|
|
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_BLUE%Standalone Application%RESET%
|
|
echo.
|
|
echo 9) %L_RED%Exit/Quit%RESET%
|
|
echo.
|
|
echo.
|
|
set /p UserOption=" Enter your choice: "
|
|
|
|
if "%UserOption%"=="1" goto WebUIMenu
|
|
if "%UserOption%"=="2" goto StandaloneMenu
|
|
if "%UserOption%"=="9" goto End
|
|
goto MainMenu
|
|
|
|
:WebUIMenu
|
|
cls
|
|
echo.
|
|
echo %L_BLUE%TEXT-GENERATION-WEBUI SETUP%RESET%
|
|
echo.
|
|
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 BASE REQUIREMENTS
|
|
echo 1) Apply/Re-Apply the requirements for %L_GREEN%Text-generation-webui%RESET%.
|
|
echo.
|
|
echo OPTIONAL
|
|
echo 2) Git Pull the latest AllTalk updates from Github
|
|
echo.
|
|
echo DEEPSPEED FOR %L_YELLOW=%PyTorch 2.1.x%RESET%
|
|
echo 4) Install DeepSpeed v11.2 for CUDA %L_GREEN%11.8%RESET% and Python-3.11.x and %L_YELLOW%PyTorch 2.1.x%RESET%.
|
|
echo 5) Install DeepSpeed v11.2 for CUDA %L_GREEN%12.1%RESET% and Python-3.11.x and %L_YELLOW%PyTorch 2.1.x%RESET%.
|
|
echo.
|
|
echo DEEPSPEED FOR %L_YELLOW=%PyTorch 2.2.x%RESET% (March 2024 builds of Text-gen-webui and later)
|
|
echo 6) Install DeepSpeed v14.0 for CUDA %L_GREEN%12.1%RESET% and Python-3.11.x and %L_YELLOW%PyTorch 2.2.x%RESET%.
|
|
echo 7) Install DeepSpeed v14.0 for CUDA %L_GREEN%11.8%RESET% and Python-3.11.x and %L_YELLOW%PyTorch 2.2.x%RESET%.
|
|
echo.
|
|
echo U) Uninstall DeepSpeed.
|
|
echo.
|
|
echo. OTHER
|
|
echo 8) Generate a diagnostics file.
|
|
echo.
|
|
echo 9) %L_RED%Exit/Quit%RESET%
|
|
echo.
|
|
set /p WebUIOption="Enter your choice: "
|
|
if "%WebUIOption%"=="1" goto InstallNvidiaTextGen
|
|
if "%WebUIOption%"=="2" goto TGGitpull
|
|
if "%WebUIOption%"=="4" goto InstallDeepSpeed118TextGen
|
|
if "%WebUIOption%"=="5" goto InstallDeepSpeed121TextGen
|
|
if "%WebUIOption%"=="6" goto InstallDeepSpeed121TextGenPytorch221
|
|
if "%WebUIOption%"=="7" goto InstallDeepSpeed118TextGenPytorch221
|
|
if "%WebUIOption%"=="U" goto UnInstallDeepSpeed
|
|
if "%WebUIOption%"=="u" goto UnInstallDeepSpeed
|
|
if "%WebUIOption%"=="8" goto GenerateDiagsTextGen
|
|
if "%WebUIOption%"=="9" goto End
|
|
goto WebUIMenu
|
|
|
|
:StandaloneMenu
|
|
cls
|
|
echo.
|
|
echo %L_BLUE%ALLTALK STANDALONE APPLICATION SETUP%RESET%
|
|
echo.
|
|
echo BASE REQUIREMENTS
|
|
echo 1) Install AllTalk as a Standalone Application
|
|
echo.
|
|
echo OPTIONAL
|
|
echo 2) Git Pull the latest AllTalk updates from Github
|
|
echo 3) Re-Apply/Update the requirements file
|
|
echo 4) Delete AllTalk's custom Python environment
|
|
echo 5) Purge the PIP cache
|
|
echo.
|
|
echo. OTHER
|
|
echo 8) Generate a diagnostics file
|
|
echo.
|
|
echo 9) %L_RED%Exit/Quit%RESET%
|
|
echo.
|
|
set /p StandaloneOption="Enter your choice: "
|
|
if "%StandaloneOption%"=="1" goto InstallCustomStandalone
|
|
if "%StandaloneOption%"=="2" goto STGitpull
|
|
if "%StandaloneOption%"=="3" goto STReapplyrequirements
|
|
if "%StandaloneOption%"=="4" goto STDeleteCustomStandalone
|
|
if "%StandaloneOption%"=="5" goto STPurgepipcache
|
|
if "%StandaloneOption%"=="8" goto GenerateDiagsStandalone
|
|
if "%StandaloneOption%"=="9" goto EndStandalone
|
|
goto StandaloneMenu
|
|
|
|
:InstallNvidiaTextGen
|
|
pip install -r system\requirements\requirements_textgen.txt
|
|
pip install -r system\requirements\requirements_textgen2.txt
|
|
pip install -r system\requirements\requirements_parler.txt
|
|
echo ** Faiss **
|
|
call "%CONDA_ROOT_PREFIX%\Scripts\conda.exe" install -y pytorch::faiss-cpu
|
|
echo ** FFmpeg **
|
|
call "%CONDA_ROOT_PREFIX%\Scripts\conda.exe" install -y conda-forge::ffmpeg
|
|
if %ERRORLEVEL% neq 0 (
|
|
echo.
|
|
echo There was an error installing the requirements.
|
|
echo Have you started your Text-gen-webui Python environment
|
|
echo with cmd_{yourOS} before running atsetup.bat?
|
|
echo Press any key to return to the menu.
|
|
echo.
|
|
pause
|
|
goto WebUIMenu
|
|
)
|
|
Echo.
|
|
echo.
|
|
Echo Requirements installed successfully.
|
|
Echo.
|
|
pause
|
|
goto WebUIMenu
|
|
|
|
:TGGitpull
|
|
git pull
|
|
if %ERRORLEVEL% neq 0 (
|
|
echo.
|
|
echo There was an error pulling from Github.
|
|
echo Press any key to return to the menu.
|
|
echo.
|
|
pause
|
|
goto WebUIMenu
|
|
)
|
|
Echo.
|
|
echo AllTalk Updated from Github. Please re-apply
|
|
echo the latest requirements file. Option 1
|
|
Echo.
|
|
pause
|
|
goto WebUIMenu
|
|
|
|
:InstallDeepSpeed118TextGen
|
|
echo Downloading DeepSpeed...
|
|
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 (
|
|
echo.
|
|
echo Failed to download DeepSpeed wheel file.
|
|
echo Please check your internet connection or try again later.
|
|
echo Press any key to return to the menu.
|
|
echo.
|
|
pause
|
|
goto WebUIMenu
|
|
)
|
|
echo DeepSpeed wheel file downloaded successfully.
|
|
echo Installing DeepSpeed...
|
|
pip install deepspeed-0.11.2+cuda118-cp311-cp311-win_amd64.whl
|
|
if %ERRORLEVEL% neq 0 (
|
|
echo.
|
|
echo Failed to install DeepSpeed.
|
|
echo Please check if the wheel file is compatible with your system.
|
|
echo Press any key to return to the menu.
|
|
echo.
|
|
pause
|
|
goto WebUIMenu
|
|
)
|
|
Echo.
|
|
echo.
|
|
Echo DeepSpeed installed successfully.
|
|
Echo.
|
|
del deepspeed-0.11.2+cuda118-cp311-cp311-win_amd64.whl
|
|
pause
|
|
goto WebUIMenu
|
|
|
|
:InstallDeepSpeed121TextGen
|
|
echo Downloading DeepSpeed...
|
|
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 (
|
|
echo.
|
|
echo Failed to download DeepSpeed wheel file.
|
|
echo Please check your internet connection or try again later.
|
|
echo Press any key to return to the menu.
|
|
echo.
|
|
pause
|
|
goto WebUIMenu
|
|
)
|
|
echo DeepSpeed wheel file downloaded successfully.
|
|
echo Installing DeepSpeed...
|
|
pip install deepspeed-0.11.2+cuda121-cp311-cp311-win_amd64.whl
|
|
if %ERRORLEVEL% neq 0 (
|
|
echo.
|
|
echo Failed to install DeepSpeed.
|
|
echo Please check if the wheel file is compatible with your system.
|
|
echo Press any key to return to the menu.
|
|
echo.
|
|
pause
|
|
goto WebUIMenu
|
|
)
|
|
Echo.
|
|
echo.
|
|
Echo DeepSpeed installed successfully.
|
|
Echo.
|
|
del deepspeed-0.11.2+cuda121-cp311-cp311-win_amd64.whl
|
|
pause
|
|
goto WebUIMenu
|
|
|
|
:InstallDeepSpeed121TextGenPytorch221
|
|
echo Downloading DeepSpeed...
|
|
curl -LO https://github.com/erew123/alltalk_tts/releases/download/DeepSpeed-14.0/deepspeed-0.14.0+ce78a63-cp311-cp311-win_amd64.whl
|
|
if %ERRORLEVEL% neq 0 (
|
|
echo.
|
|
echo Failed to download DeepSpeed wheel file.
|
|
echo Please check your internet connection or try again later.
|
|
echo Press any key to return to the menu.
|
|
echo.
|
|
pause
|
|
goto WebUIMenu
|
|
)
|
|
echo DeepSpeed wheel file downloaded successfully.
|
|
echo Installing DeepSpeed...
|
|
pip install deepspeed-0.14.0+ce78a63-cp311-cp311-win_amd64.whl
|
|
if %ERRORLEVEL% neq 0 (
|
|
echo.
|
|
echo Failed to install DeepSpeed.
|
|
echo Please check if the wheel file is compatible with your system.
|
|
echo Press any key to return to the menu.
|
|
echo.
|
|
pause
|
|
goto WebUIMenu
|
|
)
|
|
Echo.
|
|
echo.
|
|
Echo DeepSpeed installed successfully.
|
|
Echo.
|
|
del deepspeed-0.14.0+ce78a63-cp311-cp311-win_amd64.whl
|
|
pause
|
|
goto WebUIMenu
|
|
|
|
:InstallDeepSpeed118TextGenPytorch221
|
|
echo Downloading DeepSpeed...
|
|
curl -LO https://github.com/erew123/alltalk_tts/releases/download/DeepSpeed-14.0/deepspeed-0.14.0+cu118-cp311-cp311-win_amd64.whl
|
|
if %ERRORLEVEL% neq 0 (
|
|
echo.
|
|
echo Failed to download DeepSpeed wheel file.
|
|
echo Please check your internet connection or try again later.
|
|
echo Press any key to return to the menu.
|
|
echo.
|
|
pause
|
|
goto WebUIMenu
|
|
)
|
|
echo DeepSpeed wheel file downloaded successfully.
|
|
echo Installing DeepSpeed...
|
|
pip install deepspeed-0.14.0+cu118-cp311-cp311-win_amd64.whl
|
|
if %ERRORLEVEL% neq 0 (
|
|
echo.
|
|
echo Failed to install DeepSpeed.
|
|
echo Please check if the wheel file is compatible with your system.
|
|
echo Press any key to return to the menu.
|
|
echo.
|
|
pause
|
|
goto WebUIMenu
|
|
)
|
|
Echo.
|
|
echo.
|
|
Echo DeepSpeed installed successfully.
|
|
Echo.
|
|
del deepspeed-0.14.0+cu118-cp311-cp311-win_amd64.whl
|
|
pause
|
|
goto WebUIMenu
|
|
|
|
:UnInstallDeepSpeed
|
|
pip uninstall deepspeed
|
|
if %ERRORLEVEL% neq 0 (
|
|
echo.
|
|
echo There was an error uninstalling DeepSpeed.
|
|
echo Press any key to return to the menu.
|
|
echo.
|
|
pause
|
|
goto WebUIMenu
|
|
)
|
|
echo.
|
|
Echo DeepSpeed uninstalled successfully.
|
|
Echo.
|
|
pause
|
|
goto WebUIMenu
|
|
|
|
:GenerateDiagsTextGen
|
|
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 %L_GREEN%cmd_windows.bat%RESET%?
|
|
echo.
|
|
pause
|
|
goto WebUIMenu
|
|
)
|
|
Echo.
|
|
echo.
|
|
Echo Diagnostics.log generated. Please scroll up to look over the log.
|
|
Echo.
|
|
pause
|
|
goto WebUIMenu
|
|
|
|
:InstallCustomStandalone
|
|
set PATH=%PATH%;%SystemRoot%\system32
|
|
|
|
@rem Check if curl is available
|
|
curl --version >nul 2>&1
|
|
if "%ERRORLEVEL%" NEQ "0" (
|
|
echo curl is not available on this system. Please install curl then re-run the script https://curl.se/
|
|
echo or perform a manual installation of a Conda Python environment.
|
|
goto end
|
|
)
|
|
echo "%CD%"| findstr /C:" " >nul && echo This script relies on Miniconda which can not be silently installed under a path with spaces. Please correct your folder names and re-try installation. && goto exit
|
|
|
|
@rem Check for special characters in installation path
|
|
set "SPCHARMESSAGE=WARNING: Special characters were detected in the installation path! This can cause the installation to fail!"
|
|
echo "%CD%"| findstr /R /C:"[!#\$%&()\*+,;<=>?@\[\]\^`{|}~]" >nul && (
|
|
call :PrintBigMessage %SPCHARMESSAGE%
|
|
)
|
|
set SPCHARMESSAGE=
|
|
|
|
@rem fix failed install when installing to a separate drive
|
|
set TMP=%cd%\alltalk_environment
|
|
set TEMP=%cd%\alltalk_environment
|
|
|
|
@rem deactivate existing conda envs as needed to avoid conflicts
|
|
(call conda deactivate && call conda deactivate && call conda deactivate) 2>nul
|
|
|
|
@rem config
|
|
set INSTALL_DIR=%cd%\alltalk_environment
|
|
set CONDA_ROOT_PREFIX=%cd%\alltalk_environment\conda
|
|
set INSTALL_ENV_DIR=%cd%\alltalk_environment\env
|
|
set MINICONDA_DOWNLOAD_URL=https://repo.anaconda.com/miniconda/Miniconda3-py311_24.4.0-0-Windows-x86_64.exe
|
|
set conda_exists=F
|
|
|
|
@rem figure out whether git and conda need to be installed
|
|
call "%CONDA_ROOT_PREFIX%\_conda.exe" --version >nul 2>&1
|
|
if "%ERRORLEVEL%" EQU "0" (
|
|
set conda_exists=T
|
|
call "%CONDA_ROOT_PREFIX%\condabin\conda.bat" activate "%INSTALL_ENV_DIR%"
|
|
goto RunScript
|
|
)
|
|
|
|
@rem download and install conda if not installed
|
|
echo Downloading Miniconda from %MINICONDA_DOWNLOAD_URL% to %INSTALL_DIR%\miniconda_installer.exe
|
|
mkdir "%INSTALL_DIR%"
|
|
call curl -Lk "%MINICONDA_DOWNLOAD_URL%" > "%INSTALL_DIR%\miniconda_installer.exe" || ( echo. && echo Miniconda failed to download. && goto end )
|
|
echo Installing Miniconda to %CONDA_ROOT_PREFIX%
|
|
start /wait "" "%INSTALL_DIR%\miniconda_installer.exe" /InstallationType=JustMe /NoShortcuts=1 /AddToPath=0 /RegisterPython=0 /NoRegistry=1 /S /D=%CONDA_ROOT_PREFIX%
|
|
echo Miniconda version:
|
|
call "%CONDA_ROOT_PREFIX%\_conda.exe" --version || ( echo. && echo Miniconda not found. && goto end )
|
|
|
|
@rem create the installer env
|
|
echo Packages to install: %PACKAGES_TO_INSTALL%
|
|
call "%CONDA_ROOT_PREFIX%\_conda.exe" create --no-shortcuts -y -k --prefix "%INSTALL_ENV_DIR%" python=3.11.9 || ( echo. && echo Conda environment creation failed. && goto end )
|
|
|
|
@rem check if conda environment was actually created
|
|
if not exist "%INSTALL_ENV_DIR%\python.exe" ( echo. && echo Conda environment is empty. && goto end )
|
|
|
|
@rem activate installer env
|
|
call "%CONDA_ROOT_PREFIX%\condabin\conda.bat" activate "%INSTALL_ENV_DIR%" || ( echo. && echo Miniconda hook not found. && goto end )
|
|
rem Install required packages
|
|
echo ** PyTorch 2.2.1 **
|
|
call "%CONDA_ROOT_PREFIX%\Scripts\conda.exe" install -y pytorch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 pytorch-cuda=12.1 -c pytorch -c nvidia
|
|
echo ** Faiss **
|
|
call "%CONDA_ROOT_PREFIX%\Scripts\conda.exe" install -y pytorch::faiss-cpu
|
|
echo ** FFmpeg **
|
|
call "%CONDA_ROOT_PREFIX%\Scripts\conda.exe" install -y conda-forge::ffmpeg
|
|
echo ** Requirements file **
|
|
pip install -r system\requirements\requirements_standalone.txt
|
|
echo ** Gradio Update **
|
|
pip install --upgrade gradio==4.32.2
|
|
echo ** DeepSpeed **
|
|
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+ce78a63-cp311-cp311-win_amd64.whl
|
|
del deepspeed-0.14.0+ce78a63-cp311-cp311-win_amd64.whl
|
|
echo Installing Parler...
|
|
pip install -r system\requirements\requirements_parler.txt
|
|
echo ** Clean Environment **
|
|
call "%CONDA_ROOT_PREFIX%\Scripts\conda.exe" clean --all --force-pkgs-dirs -y
|
|
|
|
|
|
@rem Create start_environment.bat to run AllTalk environment
|
|
echo @echo off > start_environment.bat
|
|
echo cd /D "%~dp0" >> start_environment.bat
|
|
echo set CONDA_ROOT_PREFIX=%cd%\alltalk_environment\conda >> start_environment.bat
|
|
echo set INSTALL_ENV_DIR=%cd%\alltalk_environment\env >> start_environment.bat
|
|
echo call "%CONDA_ROOT_PREFIX%\condabin\conda.bat" activate "%INSTALL_ENV_DIR%" >> start_environment.bat
|
|
@rem Create start_alltalk.bat to run AllTalk
|
|
echo @echo off > start_alltalk.bat
|
|
echo cd /D "%~dp0" >> start_alltalk.bat
|
|
echo set CONDA_ROOT_PREFIX=%cd%\alltalk_environment\conda >> start_alltalk.bat
|
|
echo set INSTALL_ENV_DIR=%cd%\alltalk_environment\env >> start_alltalk.bat
|
|
echo call "%CONDA_ROOT_PREFIX%\condabin\conda.bat" activate "%INSTALL_ENV_DIR%" >> start_alltalk.bat
|
|
echo call python script.py >> start_alltalk.bat
|
|
@rem Create start_finetune.bat to run AllTalk
|
|
echo @echo off > start_finetune.bat
|
|
echo cd /D "%~dp0" >> start_finetune.bat
|
|
echo set CONDA_ROOT_PREFIX=%cd%\alltalk_environment\conda >> start_finetune.bat
|
|
echo set INSTALL_ENV_DIR=%cd%\alltalk_environment\env >> start_finetune.bat
|
|
echo call "%CONDA_ROOT_PREFIX%\condabin\conda.bat" activate "%INSTALL_ENV_DIR%" >> start_finetune.bat
|
|
echo call python finetune.py >> start_finetune.bat
|
|
echo @echo off > start_diagnostics.bat
|
|
echo cd /D "%~dp0" >> start_diagnostics.bat
|
|
echo set CONDA_ROOT_PREFIX=%cd%\alltalk_environment\conda >> start_diagnostics.bat
|
|
echo set INSTALL_ENV_DIR=%cd%\alltalk_environment\env >> start_diagnostics.bat
|
|
echo call "%CONDA_ROOT_PREFIX%\condabin\conda.bat" activate "%INSTALL_ENV_DIR%" >> start_diagnostics.bat
|
|
echo call python diagnostics.py >> start_diagnostics.bat
|
|
Echo.
|
|
Echo Run %L_YELLOW%start_alltalk.bat%RESET% to start AllTalk.
|
|
Echo Run %L_YELLOW%start_diagnostics.bat%RESET% to start the diagnostics.
|
|
Echo Run %L_YELLOW%start_finetune.bat%RESET% to start Finetuning.
|
|
Echo Run %L_YELLOW%start_environment.bat%RESET% to start the AllTalk Python environment.
|
|
Echo.
|
|
if "%1"=="-silent" goto End
|
|
pause
|
|
goto StandaloneMenu
|
|
|
|
:STDeleteCustomStandalone
|
|
@rem Check if the alltalk_environment directory exists
|
|
if not exist "%cd%\alltalk_environment\" (
|
|
echo.
|
|
echo %L_GREEN%alltalk_environment%RESET% directory does not exist. No need to delete.
|
|
echo.
|
|
pause
|
|
goto StandaloneMenu
|
|
)
|
|
@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 %L_GREEN%atsetup.bat%RESET% again and delete the environment.
|
|
call conda deactivate
|
|
:NoCondaEnvDeleteCustomStandalone
|
|
echo Deleting "alltalk_environment". Please wait.
|
|
rd /s /q "alltalk_environment"
|
|
del start_alltalk.bat
|
|
del start_environment.bat
|
|
if %ERRORLEVEL% neq 0 (
|
|
echo.
|
|
echo Failed to delete alltalk_environment folder.
|
|
echo Please make sure it is not in use and try again.
|
|
echo Worst case, manually delete the folder called 'alltalk_environment'
|
|
echo.
|
|
pause
|
|
goto StandaloneMenu
|
|
)
|
|
echo.
|
|
Echo.
|
|
echo Environment %L_GREEN%alltalk_environment%RESET% deleted. Please set up the environment again.
|
|
Echo.
|
|
pause
|
|
goto StandaloneMenu
|
|
|
|
:GenerateDiagsStandalone
|
|
cd /D "%~dp0"
|
|
set CONDA_ROOT_PREFIX=%cd%\alltalk_environment\conda
|
|
set INSTALL_ENV_DIR=%cd%\alltalk_environment\env
|
|
@rem Check if the Conda environment exists
|
|
if not exist "%INSTALL_ENV_DIR%\python.exe" (
|
|
echo.
|
|
echo The Conda environment at "%INSTALL_ENV_DIR%" does not exist.
|
|
echo Please install the environment before proceeding.
|
|
echo.
|
|
pause
|
|
goto StandaloneMenu
|
|
)
|
|
@rem Attempt to activate the Conda environment
|
|
call "%CONDA_ROOT_PREFIX%\condabin\conda.bat" activate "%INSTALL_ENV_DIR%"
|
|
if errorlevel 1 (
|
|
echo.
|
|
echo Failed to activate the Conda environment.
|
|
echo Please check your installation and try again.
|
|
echo.
|
|
pause
|
|
goto StandaloneMenu
|
|
)
|
|
@rem Run diagnostics
|
|
Python diagnostics.py
|
|
if %ERRORLEVEL% neq 0 (
|
|
echo.
|
|
echo There was an error running diagnostics.
|
|
echo Press any key to return to the menu.
|
|
echo.
|
|
pause
|
|
goto StandaloneMenu
|
|
)
|
|
Echo.
|
|
Echo.
|
|
Echo Diagnostics.log generated. Please scroll up to look over the log.
|
|
Echo.
|
|
pause
|
|
goto StandaloneMenu
|
|
|
|
:STReapplyrequirements
|
|
cd /D "%~dp0"
|
|
set CONDA_ROOT_PREFIX=%cd%\alltalk_environment\conda
|
|
set INSTALL_ENV_DIR=%cd%\alltalk_environment\env
|
|
@rem Check if the Conda environment exists
|
|
if not exist "%INSTALL_ENV_DIR%\python.exe" (
|
|
echo.
|
|
echo The Conda environment at "%INSTALL_ENV_DIR%" does not exist.
|
|
echo Please install the environment before proceeding.
|
|
echo.
|
|
pause
|
|
goto StandaloneMenu
|
|
)
|
|
@rem Attempt to activate the Conda environment
|
|
call "%CONDA_ROOT_PREFIX%\condabin\conda.bat" activate "%INSTALL_ENV_DIR%"
|
|
if errorlevel 1 (
|
|
echo.
|
|
echo Failed to activate the Conda environment.
|
|
echo Please check your installation and try again.
|
|
echo.
|
|
pause
|
|
goto StandaloneMenu
|
|
)
|
|
@rem Run Reapply requirements
|
|
echo.
|
|
echo.
|
|
echo Installing requirements. If you have problems after this, you should
|
|
echo delete the alltalk_environment and run the setup routine again.
|
|
echo.
|
|
pip install -r system\requirements\requirements_standalone.txt
|
|
pip install -r system\requirements\requirements_parler.txt
|
|
echo Installing DeepSpeed...
|
|
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.
|
|
echo Press any key to return to the menu.
|
|
echo.
|
|
pause
|
|
goto StandaloneMenu
|
|
)
|
|
Echo.
|
|
Echo.
|
|
Echo Requirements have been re-applied/updated.
|
|
Echo.
|
|
pause
|
|
goto StandaloneMenu
|
|
|
|
:STPurgepipcache
|
|
cd /D "%~dp0"
|
|
set CONDA_ROOT_PREFIX=%cd%\alltalk_environment\conda
|
|
set INSTALL_ENV_DIR=%cd%\alltalk_environment\env
|
|
@rem Check if the Conda environment exists
|
|
if not exist "%INSTALL_ENV_DIR%\python.exe" (
|
|
echo.
|
|
echo The Conda environment at "%INSTALL_ENV_DIR%" does not exist.
|
|
echo Please install the environment before proceeding.
|
|
echo.
|
|
pause
|
|
goto StandaloneMenu
|
|
)
|
|
@rem Attempt to activate the Conda environment
|
|
call "%CONDA_ROOT_PREFIX%\condabin\conda.bat" activate "%INSTALL_ENV_DIR%"
|
|
if errorlevel 1 (
|
|
echo.
|
|
echo Failed to activate the Conda environment.
|
|
echo Please check your installation and try again.
|
|
echo.
|
|
pause
|
|
goto StandaloneMenu
|
|
)
|
|
@rem Clear the PIP cache
|
|
echo.
|
|
echo Purging the PIP cache of downloaded files.
|
|
echo.
|
|
pip cache purge
|
|
if %ERRORLEVEL% neq 0 (
|
|
echo.
|
|
echo There was an error.
|
|
echo Press any key to return to the menu.
|
|
echo.
|
|
pause
|
|
goto StandaloneMenu
|
|
)
|
|
Echo.
|
|
Echo The PIP cache has been purged.
|
|
Echo.
|
|
pause
|
|
goto StandaloneMenu
|
|
|
|
:STGitpull
|
|
cd /D "%~dp0"
|
|
set CONDA_ROOT_PREFIX=%cd%\alltalk_environment\conda
|
|
set INSTALL_ENV_DIR=%cd%\alltalk_environment\env
|
|
@rem Check if the Conda environment exists
|
|
if not exist "%INSTALL_ENV_DIR%\python.exe" (
|
|
echo.
|
|
echo The Conda environment at "%INSTALL_ENV_DIR%" does not exist.
|
|
echo Please install the environment before proceeding.
|
|
echo.
|
|
pause
|
|
goto StandaloneMenu
|
|
)
|
|
@rem Attempt to activate the Conda environment
|
|
call "%CONDA_ROOT_PREFIX%\condabin\conda.bat" activate "%INSTALL_ENV_DIR%"
|
|
if errorlevel 1 (
|
|
echo.
|
|
echo Failed to activate the Conda environment.
|
|
echo Please check your installation and try again.
|
|
echo.
|
|
pause
|
|
goto StandaloneMenu
|
|
)
|
|
@rem Pull from Github
|
|
echo.
|
|
echo Pulling the latest updates. Please re-apply
|
|
echo the latest requirements file. Option 3
|
|
echo.
|
|
git pull
|
|
if %ERRORLEVEL% neq 0 (
|
|
echo.
|
|
echo There was an error pulling from Github.
|
|
echo Press any key to return to the menu.
|
|
echo.
|
|
pause
|
|
goto StandaloneMenu
|
|
)
|
|
Echo.
|
|
echo AllTalk Updated from Github. Please re-apply
|
|
echo the latest requirements file. Option 3
|
|
Echo.
|
|
pause
|
|
goto StandaloneMenu
|
|
|
|
:EndStandalone
|
|
echo Exiting AllTalk Setup Utility...
|
|
echo.
|
|
Echo Remember, after installation you can....
|
|
Echo Run %L_YELLOW%start_alltalk.bat%RESET% to start AllTalk.
|
|
Echo Run %L_YELLOW%start_finetune.bat%RESET% to start Finetuning.
|
|
Echo Run %L_YELLOW%start_environment.bat%RESET% to start the AllTalk Python environment.
|
|
Echo.
|
|
exit /b
|
|
|
|
:End
|
|
echo Exiting AllTalk Setup Utility...
|
|
exit /b
|