mirror of
https://github.com/mudler/LocalAI.git
synced 2025-04-13 06:03:09 +00:00
fix(diffusers): pin torch and torchvision (#1592)
Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
This commit is contained in:
parent
63ee689f21
commit
0c0bc18c94
@ -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
|
||||
|
@ -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
|
||||
|
Loading…
x
Reference in New Issue
Block a user