From c22c9cd142b458504629705c9a2d81e2a8ad7858 Mon Sep 17 00:00:00 2001
From: litagin02 <litagin02@gmail.com>
Date: Fri, 2 Feb 2024 22:09:57 +0900
Subject: [PATCH] Fix torch 2.2.0 bug to use 2.1.2

---
 requirements.txt                     | 2 +-
 scripts/Install-Style-Bert-VITS2.bat | 9 ++++-----
 2 files changed, 5 insertions(+), 6 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index e1dbf67..4cb8bad 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -24,6 +24,6 @@ requests
 safetensors
 scipy
 tensorboard
-torch # For users without GPU
+torch==2.1.2 # For users without GPU
 transformers
 umap-learn
\ No newline at end of file
diff --git a/scripts/Install-Style-Bert-VITS2.bat b/scripts/Install-Style-Bert-VITS2.bat
index f6c9b9a..6e99230 100644
--- a/scripts/Install-Style-Bert-VITS2.bat
+++ b/scripts/Install-Style-Bert-VITS2.bat
@@ -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の起動