dearwolf/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2025-05-20 10:35:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 21

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

Browse source 
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
Download patch file Download diff file Expand all files Collapse all files

5
backend/python/diffusers/install.sh
View file

@ -11,9 +11,4 @@ if [ "x${BUILD_PROFILE}" == "xintel" ]; then
EXTRA_PIP_INSTALL_FLAGS+=" --upgrade --index-strategy=unsafe-first-match" EXTRA_PIP_INSTALL_FLAGS+=" --upgrade --index-strategy=unsafe-first-match"
fi 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 installRequirements

6
backend/python/diffusers/requirements-hipblas.txt
View file

@ -1,6 +1,6 @@
--extra-index-url https://download.pytorch.org/whl/nightly/rocm6.1/ --extra-index-url https://download.pytorch.org/whl/rocm6.0
torch torch==2.3.1+rocm6.0
torchvision torchvision==0.18.1+rocm6.0
diffusers diffusers
opencv-python opencv-python
transformers transformers