Fix torch 2.2.0 bug to use 2.1.2

This commit is contained in:
litagin02
2024-02-02 22:09:57 +09:00
parent ceee581575
commit c22c9cd142
2 changed files with 5 additions and 6 deletions

View File

@@ -24,6 +24,6 @@ requests
safetensors
scipy
tensorboard
torch # For users without GPU
torch==2.1.2 # For users without GPU
transformers
umap-learn

View File

@@ -33,7 +33,7 @@ xcopy /QSY .\Style-Bert-VITS2-master\ .\Style-Bert-VITS2\
rmdir /s /q Style-Bert-VITS2-master
echo ----------------------------------------
echo Python環境の構築を開始します。
echo Setup Python and Virtual Environment
echo ----------------------------------------
@REM Pythonと仮想環境のセットアップを呼び出す仮想環境が有効化されて戻ってくる
@@ -41,14 +41,14 @@ call Style-Bert-VITS2\scripts\Setup-Python.bat ..\..\lib\python ..\venv
if %errorlevel% neq 0 ( popd & exit /b %errorlevel% )
@REM 依存関係インストール
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
pip install torch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu118
if %errorlevel% neq 0 ( pause & popd & exit /b %errorlevel% )
pip install -r Style-Bert-VITS2\requirements.txt
if %errorlevel% neq 0 ( pause & popd & exit /b %errorlevel% )
echo ----------------------------------------
echo 環境構築が完了しました。モデルのダウンロードを開始します。
echo Environment setup is complete. Start downloading the model.
echo ----------------------------------------
@REM Style-Bert-VITS2フォルダに移動
@@ -58,8 +58,7 @@ pushd Style-Bert-VITS2
python initialize.py
echo ----------------------------------------
echo モデルのダウンロードが完了し、インストールが完了しました!
echo 音声合成のWebUIを起動します。
echo Model download is complete. Start the WebUI of the voice synthesis.
echo ----------------------------------------
@REM 音声合成WebUIの起動