huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-10-09 23:15:23 +00:00
Code Issues Packages Projects Releases Wiki Activity
4dca35fc62
text-generation-inference/server
History
drbh 4dca35fc62 feat: adjust attn weight loading logic (#1975) 
This PR updates `load_attention` to prefer loading specific attention
based on the model type. Additionally there were two cases where
`TensorParallelColumnLinear.load_multi` was called and this reduces it
to a single path
2024-09-24 03:16:16 +00:00
..
custom_kernels chore: add pre-commit (#1569) 2024-04-24 15:32:02 +03:00
exllama_kernels MI300 compatibility (#1764) 2024-07-17 05:36:58 +00:00
exllamav2_kernels chore: add pre-commit (#1569) 2024-04-24 15:32:02 +03:00
tests Aligin the source code with main branch 2.0.4 2024-09-24 03:06:55 +00:00
text_generation_server feat: adjust attn weight loading logic (#1975) 2024-09-24 03:16:16 +00:00
.gitignore Impl simple mamba model (#1480) 2024-04-23 11:45:11 +03:00
Makefile Aligin the source code with main branch 2.0.4 2024-09-24 03:06:55 +00: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 MI300 compatibility (#1764) 2024-07-17 05:36:58 +00:00
Makefile-selective-scan chore: add pre-commit (#1569) 2024-04-24 15:32:02 +03:00
Makefile-vllm MI300 compatibility (#1764) 2024-07-17 05:36:58 +00:00
poetry.lock Aligin the source code with main branch 2.0.4 2024-09-24 03:06:55 +00:00
pyproject.toml Aligin the source code with main branch 2.0.4 2024-09-24 03:06:55 +00:00
README.md chore: add pre-commit (#1569) 2024-04-24 15:32:02 +03:00
requirements_cuda.txt Modifing the version number. 2024-07-17 05:36:58 +00:00
requirements_rocm.txt Modifing the version number. 2024-07-17 05:36:58 +00:00

README.md

Text Generation Inference Python gRPC Server

A Python gRPC server for Text Generation Inference

Install

make install

Run

make run-dev