From 0c0bc18c94e8c000c7d2d47cb8bdac8b4c73e5cc Mon Sep 17 00:00:00 2001 From: Ettore Di Giacinto Date: Sat, 10 Aug 2024 10:10:47 +0200 Subject: [PATCH] fix(diffusers): pin torch and torchvision (#1592) Signed-off-by: Ettore Di Giacinto --- backend/python/diffusers/install.sh | 5 ----- backend/python/diffusers/requirements-hipblas.txt | 6 +++--- 2 files changed, 3 insertions(+), 8 deletions(-) diff --git a/backend/python/diffusers/install.sh b/backend/python/diffusers/install.sh index b0b46a86e189..36443ef1c559 100755 --- a/backend/python/diffusers/install.sh +++ b/backend/python/diffusers/install.sh @@ -11,9 +11,4 @@ if [ "x${BUILD_PROFILE}" == "xintel" ]; then EXTRA_PIP_INSTALL_FLAGS+=" --upgrade --index-strategy=unsafe-first-match" fi -# hipblas builds from nightly that needs pre-releases to work -if [ "x${BUILD_PROFILE}" == "xhipblas" ]; then - EXTRA_PIP_INSTALL_FLAGS+=" --prerelease=allow" -fi - installRequirements diff --git a/backend/python/diffusers/requirements-hipblas.txt b/backend/python/diffusers/requirements-hipblas.txt index 8c4c070cd68c..fc9ea3b43c21 100644 --- a/backend/python/diffusers/requirements-hipblas.txt +++ b/backend/python/diffusers/requirements-hipblas.txt @@ -1,6 +1,6 @@ ---extra-index-url https://download.pytorch.org/whl/nightly/rocm6.1/ -torch -torchvision +--extra-index-url https://download.pytorch.org/whl/rocm6.0 +torch==2.3.1+rocm6.0 +torchvision==0.18.1+rocm6.0 diffusers opencv-python transformers