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

* 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:
authored and GitHub committed 2024-08-05 16:38:33 +00:00
1 parent f15a93b19b
commit ed322bf59f
45 files changed
+69 -12

No files matched your search

@@ -0,0 +1,2 @@
--extra-index-url https://download.pytorch.org/whl/cu118
torch
@@ -0,0 +1 @@
torch
@@ -8,6 +8,5 @@ opencv-python
pillow
protobuf
sentencepiece
torch
transformers
certifi