huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-22 15:32:08 +00:00
Code Issues Packages Projects Releases Wiki Activity
4b376b30f1
text-generation-inference/server
History
fxmarty 4b376b30f1 GPTQ support on ROCm (#1489) 
Tested with
```
CUDA_VISIBLE_DEVICES=0 text-generation-launcher --model-id TheBloke/Llama-2-7B-Chat-GPTQ --quantize gptq
EXLLAMA_VERSION=1 CUDA_VISIBLE_DEVICES=0 text-generation-launcher --model-id TheBloke/Llama-2-7B-Chat-GPTQ --quantize gptq
CUDA_VISIBLE_DEVICES="0,1" text-generation-launcher --model-id TheBloke/Llama-2-7B-Chat-GPTQ --quantize gptq
```

all with good and identical results on MI210.

---------

Co-authored-by: Felix Marty <felix@hf.co>
Co-authored-by: OlivierDehaene <olivier@huggingface.co>
Co-authored-by: OlivierDehaene <23298448+OlivierDehaene@users.noreply.github.com>
2024-04-22 15:38:50 +03:00
..
custom_kernels Add RoCm support (#1243) 2023-11-27 14:08:12 +01:00
exllama_kernels GPTQ support on ROCm (#1489) 2024-04-22 15:38:50 +03:00
exllamav2_kernels GPTQ support on ROCm (#1489) 2024-04-22 15:38:50 +03:00
tests Fixing non divisible embeddings. (#1476) 2024-04-22 12:48:59 +03:00
text_generation_server GPTQ support on ROCm (#1489) 2024-04-22 15:38:50 +03:00
.gitignore Support eetq weight only quantization (#1068) 2023-09-27 11:42:57 +02:00
Makefile feat: mixtral (#1328) 2024-04-18 12:39:52 +00:00
Makefile-awq Add AWQ quantization inference support (#1019) (#1054) 2023-09-25 15:31:27 +02:00
Makefile-eetq Support eetq weight only quantization (#1068) 2023-09-27 11:42:57 +02:00
Makefile-flash-att Add RoCm support (#1243) 2023-11-27 14:08:12 +01:00
Makefile-flash-att-v2 make install-flash-attn-v2-cuda should work like make install-flash-attn-v2 used to work. (#1294) 2023-11-28 16:28:40 +01:00
Makefile-vllm Speculative (#1308) 2024-04-18 12:39:39 +00:00
poetry.lock feat: relax mistral requirements (#1351) 2024-04-19 14:50:24 +03:00
pyproject.toml v1.3.4 2024-04-22 09:08:34 +03:00
README.md feat(router): refactor API and add openAPI schemas (#53) 2023-02-03 12:43:37 +01:00
requirements_common.txt Add RoCm support (#1243) 2023-11-27 14:08:12 +01:00
requirements_cuda.txt feat: relax mistral requirements (#1351) 2024-04-19 14:50:24 +03:00
requirements_rocm.txt feat: relax mistral requirements (#1351) 2024-04-19 14:50:24 +03:00
requirements.txt Update dependencies (#69) 2024-02-25 13:07:47 +01:00

README.md

Text Generation Inference Python gRPC Server

A Python gRPC server for Text Generation Inference

Install

make install

Run

make run-dev