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

fix: ensure correct version of torch is always installed based on BUILD_TYPE(#2890)

Browse source 
* fix: ensure correct version of torch is always installed based on BUILD_TYPE

Signed-off-by: Chris Jowett <421501+cryptk@users.noreply.github.com>

* Move causal-conv1d installation to build_types

Signed-off-by: mudler <mudler@localai.io>

* Move mamba-ssd install to build-type requirements.txt

Signed-off-by: mudler <mudler@localai.io>

---------

Signed-off-by: Chris Jowett <421501+cryptk@users.noreply.github.com>
Signed-off-by: mudler <mudler@localai.io>
Co-authored-by: Ettore Di Giacinto <mudler@users.noreply.github.com>
Co-authored-by: mudler <mudler@localai.io>
This commit is contained in:
cryptk 2024-08-05 11:38:33 -05:00 • committed by GitHub
parent f15a93b19b
commit ed322bf59f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
45 changed files with 69 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

2
backend/python/mamba/requirements-after.txt Normal file
View file

@ -0,0 +1,2 @@
causal-conv1d==1.4.0
mamba-ssm==2.2.2

1
backend/python/mamba/requirements-cpu.txt Normal file
View file

@ -0,0 +1 @@
torch

2
backend/python/mamba/requirements-cublas11.txt Normal file
View file

@ -0,0 +1,2 @@
--extra-index-url https://download.pytorch.org/whl/cu118
torch

1
backend/python/mamba/requirements-cublas12.txt Normal file
View file

@ -0,0 +1 @@
torch

3
backend/python/mamba/requirements-install.txt
View file

@ -3,5 +3,4 @@
# https://github.com/Dao-AILab/causal-conv1d/issues/24
packaging
setuptools
wheel
torch==2.3.1
wheel

2
backend/python/mamba/requirements.txt
View file

@ -1,5 +1,3 @@
causal-conv1d==1.4.0
mamba-ssm==2.2.2
grpcio==1.65.1
protobuf
certifi