LocalAI/backend/python/mamba
Ettore Di Giacinto 5d1018495f
feat(intel): add diffusers/transformers support (#1746)
* feat(intel): add diffusers support

* try to consume upstream container image

* Debug

* Manually install deps

* Map transformers/hf cache dir to modelpath if not specified

* fix(compel): update initialization, pass by all gRPC options

* fix: add dependencies, implement transformers for xpu

* base it from the oneapi image

* Add pillow

* set threads if specified when launching the API

* Skip conda install if intel

* defaults to non-intel

* ci: add to pipelines

* prepare compel only if enabled

* Skip conda install if intel

* fix cleanup

* Disable compel by default

* Install torch 2.1.0 with Intel

* Skip conda on some setups

* Detect python

* Quiet output

* Do not override system python with conda

* Prefer python3

* Fixups

* exllama2: do not install without conda (overrides pytorch version)

* exllama/exllama2: do not install if not using cuda

* Add missing dataset dependency

* Small fixups, symlink to python, add requirements

* Add neural_speed to the deps

* correctly handle model offloading

* fix: device_map == xpu

* go back at calling python, fixed at dockerfile level

* Exllama2 restricted to only nvidia gpus

* Tokenizer to xpu
2024-03-07 14:37:45 +01:00
..
backend_mamba.py feat(extra-backends): Improvements, adding mamba example (#1618) 2024-01-20 17:56:08 +01:00
backend_pb2_grpc.py feat: 🐍 add mamba support (#1589) 2024-01-19 23:42:50 +01:00
backend_pb2.py Bump vLLM version + more options when loading models in vLLM (#1782) 2024-03-01 22:48:53 +01:00
install.sh feat(intel): add diffusers/transformers support (#1746) 2024-03-07 14:37:45 +01:00
Makefile feat: 🐍 add mamba support (#1589) 2024-01-19 23:42:50 +01:00
README.md feat: 🐍 add mamba support (#1589) 2024-01-19 23:42:50 +01:00
run.sh feat: 🐍 add mamba support (#1589) 2024-01-19 23:42:50 +01:00
test_backend_mamba.py feat: 🐍 add mamba support (#1589) 2024-01-19 23:42:50 +01:00
test.sh feat: 🐍 add mamba support (#1589) 2024-01-19 23:42:50 +01:00

Creating a separate environment for the mamba project

make mamba