LocalAI/requirements-cublas11.txt at ed322bf59f0dcc4d3c7329c829e52e6cbcd02291 - LocalAI - Gitea

ExternalVendorCode/LocalAI

mirror of https://github.com/mudler/LocalAI.git synced 2024-12-19 12:47:54 +00:00

cryptk ed322bf59f

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>

2024-08-05 16:38:33 +00:00

3 lines

73 B

Plaintext

Raw Blame History

	`--extra-index-url https://download.pytorch.org/whl/cu118`
	`torch`
	`flash-attn`