text-generation-inference/server
Jimin Ha 1023de8048
Add flash_attention argument options for Mistral (#145)
Co-authored-by: Karol Damaszke <karol.damaszke@intel.com>
Co-authored-by: Karol Damaszke <kdamaszke@habana.ai>
2024-05-27 20:00:42 +02:00
..
custom_kernels chore: add pre-commit (#1569) 2024-04-24 15:32:02 +03:00
exllama_kernels chore: add pre-commit (#1569) 2024-04-24 15:32:02 +03:00
exllamav2_kernels chore: add pre-commit (#1569) 2024-04-24 15:32:02 +03:00
tests Add grammar support (#140) 2024-05-20 11:16:34 +02:00
text_generation_server Add flash_attention argument options for Mistral (#145) 2024-05-27 20:00:42 +02:00
.gitignore Impl simple mamba model (#1480) 2024-04-23 11:45:11 +03:00
dill-0.3.7-patch.sh Hgraph dill patch (#131) 2024-04-26 11:08:15 +02:00
dill-0.3.8-patch.sh A patch to address HPU Graphs issue with DILL 2024-05-06 09:15:46 +03:00
Makefile fix: fix CohereForAI/c4ai-command-r-plus (#1707) 2024-04-25 17:51:35 +03:00
Makefile-awq chore: add pre-commit (#1569) 2024-04-24 15:32:02 +03:00
Makefile-eetq Upgrade EETQ (Fixes the cuda graphs). (#1729) 2024-04-25 17:58:27 +03:00
Makefile-flash-att chore: add pre-commit (#1569) 2024-04-24 15:32:02 +03:00
Makefile-flash-att-v2 fix: fix CohereForAI/c4ai-command-r-plus (#1707) 2024-04-25 17:51:35 +03:00
Makefile-selective-scan chore: add pre-commit (#1569) 2024-04-24 15:32:02 +03:00
Makefile-vllm fix: fix CohereForAI/c4ai-command-r-plus (#1707) 2024-04-25 17:51:35 +03:00
poetry.lock Update optimum-habana version to 1.11.1 (#138) (#142) 2024-05-14 14:13:31 +02:00
pyproject.toml Update optimum-habana version to 1.11.1 (#138) (#142) 2024-05-14 14:13:31 +02:00
README.md chore: add pre-commit (#1569) 2024-04-24 15:32:02 +03:00
requirements_cuda.txt Update libraries (#1713) 2024-04-25 17:53:45 +03:00
requirements_rocm.txt Update libraries (#1713) 2024-04-25 17:53:45 +03:00
requirements.txt Update optimum-habana version to 1.11.1 (#138) (#142) 2024-05-14 14:13:31 +02:00

Text Generation Inference Python gRPC Server

A Python gRPC server for Text Generation Inference

Install

make install

Run

make run-dev