huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-24 16:32:12 +00:00
Code Issues Packages Projects Releases Wiki Activity
67347950b7
text-generation-inference/server
History
Nicolas Patry 67347950b7
feat(server): Implements sharding for non divisible vocab_size. (#583) 
- The code is relatively easy (just disable the checks on Embedding and
Head)

This cannot be done in the same easy fashion for hidden_dim/head_dim.
It's relatively easy on some models (classic MHA) but it would make the
other
models (MQA) much more complex, and GPTQ quantization another quite
hairy piece
of code.
2023-07-12 16:43:31 +02:00
..
custom_kernels feat(server): Rework model loading (#344) 2023-06-08 14:51:52 +02:00
tests fix(server): harden the weights choice to save on disk. (#561) 2023-07-07 14:50:12 +02:00
text_generation_server feat(server): Implements sharding for non divisible vocab_size. (#583) 2023-07-12 16:43:31 +02:00
.gitignore feat(clients): Python client (#103) 2023-03-07 18:52:22 +01:00
Makefile fix: Update server/Makefile to include Makefile-vllm (#520) 2023-07-04 09:39:25 +02:00
Makefile-flash-att feat(server): use latest flash attention commit (#543) 2023-07-04 20:23:55 +02:00
Makefile-vllm feat(server): add paged attention to flash models (#516) 2023-06-30 19:09:59 +02:00
poetry.lock feat(server): use latest flash attention commit (#543) 2023-07-04 20:23:55 +02:00
pyproject.toml v0.9.1 (#558) 2023-07-06 16:05:42 +02:00
README.md feat(router): refactor API and add openAPI schemas (#53) 2023-02-03 12:43:37 +01:00
requirements.txt feat(server): use latest flash attention commit (#543) 2023-07-04 20:23:55 +02:00

README.md

Text Generation Inference Python gRPC Server

A Python gRPC server for Text Generation Inference

Install

make install

Run

make run-dev