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
9a12941bef
text-generation-inference/server
History
Ubuntu 9a12941bef [WIP] Inference support for GPTQ (llama at least) 
Let's start discussing implementation.

- Need to expose the quantization scripts (either included here or add
  doc on how to use https://github.com/qwopqwop200/GPTQ-for-LLaMa)
- Make sure GPTQ works for multiple models (priority to Falcon).

Currently it means that every place we use `get_{tensor|sharded}` to
check for quantization.

My idea is to reintegrate as much as possible into `utils/layer.py` by
expanding `load_multi` to be a bit more generic.
This might require some thinking, but ultimately the
`qweight,qzeros,scales,g_idx` should be in a single place, and
independant of bias presence.
2023-06-14 09:42:55 +02:00
..
custom_kernels feat(server): Rework model loading (#344) 2023-06-08 14:51:52 +02:00
tests feat(server): Rework model loading (#344) 2023-06-08 14:51:52 +02:00
text_generation_server [WIP] Inference support for GPTQ (llama at least) 2023-06-14 09:42:55 +02:00
.gitignore feat(clients): Python client (#103) 2023-03-07 18:52:22 +01:00
Makefile feat(server): Rework model loading (#344) 2023-06-08 14:51:52 +02:00
Makefile-flash-att feat(server): pre-allocate past key values for flash causal LM (#412) 2023-06-12 18:30:29 +02:00
poetry.lock chore(sever): update requirements (#357) 2023-05-23 18:03:22 +02:00
pyproject.toml feat(server): Rework model loading (#344) 2023-06-08 14:51:52 +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): Rework model loading (#344) 2023-06-08 14:51:52 +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