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