text-generation-inference/server/text_generation_server
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
..
models Add sealion mpt support (#1477) 2024-04-22 15:37:05 +03:00
pb feat(server): clear cache on error (#143) 2023-03-28 11:29:35 +02:00
utils GPTQ support on ROCm (#1489) 2024-04-22 15:38:50 +03:00
__init__.py feat(clients): Python client (#103) 2023-03-07 18:52:22 +01:00
cache.py fix(server): decrease memory fragmentation (#557) 2023-07-06 14:28:33 +02:00
cli.py Fix local load for Medusa (#1420) 2024-04-22 09:30:41 +03:00
habana_quantization_env.py Add Habana copyright header (#122) 2024-04-08 18:06:21 +02:00
interceptor.py Add Habana copyright header (#122) 2024-04-08 18:06:21 +02:00
server.py fix: fix gpt-q with groupsize = -1 (#1358) 2024-04-19 15:05:50 +03:00
tgi_service.py Speculative (#1308) 2024-04-18 12:39:39 +00:00
tracing.py feat(clients): Python client (#103) 2023-03-07 18:52:22 +01:00