diff --git a/setup-directml.bat b/setup-directml.bat index 929fb5a..3762860 100755 --- a/setup-directml.bat +++ b/setup-directml.bat @@ -4,7 +4,7 @@ git submodule update --remote python -m venv venv call .\venv\Scripts\activate.bat python -m pip install --upgrade pip -python -m pip install torch torchvision torchaudio torch-directml +python -m pip install torch==1.13.1 torchvision torchaudio torch-directml python -m pip install -r .\modules\tortoise-tts\requirements.txt python -m pip install -e .\modules\tortoise-tts\ python -m pip install -r .\modules\dlas\requirements.txt diff --git a/setup-rocm.sh b/setup-rocm.sh index 1e010e2..3ac1cd4 100755 --- a/setup-rocm.sh +++ b/setup-rocm.sh @@ -7,7 +7,7 @@ python3 -m venv venv source ./venv/bin/activate python3 -m pip install --upgrade pip # just to be safe # ROCM -pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.4.2 +pip3 install torch==1.13.1 torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.2 # 5.4.2 doesn't work for me desu # install requirements python3 -m pip install -r ./modules/tortoise-tts/requirements.txt # install TorToiSe requirements python3 -m pip install -e ./modules/tortoise-tts/ # install TorToiSe