fix(diffusers): pin torch and torchvision (#1592)

Signed-off-by: Ettore Di Giacinto <[email protected]>
mudler · Aug 10, 2024 · 0c0bc18 · 0c0bc18
1 parent 63ee689
commit 0c0bc18
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 8 deletions.
diff --git 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
@@ -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