Skip to content

Commit 0c0bc18

Browse files
committed
fix(diffusers): pin torch and torchvision (#1592)
Signed-off-by: Ettore Di Giacinto <[email protected]>
1 parent 63ee689 commit 0c0bc18

File tree

2 files changed

+3
-8
lines changed

2 files changed

+3
-8
lines changed

backend/python/diffusers/install.sh

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -11,9 +11,4 @@ if [ "x${BUILD_PROFILE}" == "xintel" ]; then
1111
EXTRA_PIP_INSTALL_FLAGS+=" --upgrade --index-strategy=unsafe-first-match"
1212
fi
1313

14-
# hipblas builds from nightly that needs pre-releases to work
15-
if [ "x${BUILD_PROFILE}" == "xhipblas" ]; then
16-
EXTRA_PIP_INSTALL_FLAGS+=" --prerelease=allow"
17-
fi
18-
1914
installRequirements

backend/python/diffusers/requirements-hipblas.txt

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
1-
--extra-index-url https://download.pytorch.org/whl/nightly/rocm6.1/
2-
torch
3-
torchvision
1+
--extra-index-url https://download.pytorch.org/whl/rocm6.0
2+
torch==2.3.1+rocm6.0
3+
torchvision==0.18.1+rocm6.0
44
diffusers
55
opencv-python
66
transformers

0 commit comments

Comments
 (0)