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

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
This commit is contained in:
Ettore Di Giacinto 2024-08-10 10:10:47 +02:00
parent 63ee689f21
commit 0c0bc18c94
2 changed files with 3 additions and 8 deletions

View File

@ -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

View File

@ -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