hf_text-generation-inference/server/text_generation_server
Daniël de Kok bcb3faa1c2
Factor out sharding of packed tensors (#2059)
For Phi-3-Small I need to shard a packed QKV bias tensor, for which
I implemented the `Weights.get_packed_sharded` method. However, this
method can also replace the `Weights._get_qweight` method and the
custom sharding code from `Weights.get_weights_col_packed`.
2024-06-20 09:56:04 +02:00
..
layers Add support for GPTQ Marlin (#2052) 2024-06-14 09:45:42 +02:00
models Support exl2-quantized Qwen2 models (#2085) 2024-06-20 07:56:16 +02:00
pb chore: add pre-commit (#1569) 2024-02-16 11:58:58 +01:00
utils Factor out sharding of packed tensors (#2059) 2024-06-20 09:56:04 +02: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 ROCm and sliding windows fixes (#2033) 2024-06-10 15:09:50 +08:00
interceptor.py v2.0.0 (#1736) 2024-04-12 18:38:34 +02:00
server.py Set maximum grpc message receive size to 2GiB (#2075) 2024-06-17 16:40:44 +02:00
tracing.py feat(clients): Python client (#103) 2023-03-07 18:52:22 +01:00