huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-21 23:12:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
c0f201c9d3
text-generation-inference/server/text_generation_server
History
Daniël de Kok c0f201c9d3 Factor out sharding of packed tensors 
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-12 16:20:51 +02:00
..
layers Add Phi-3 medium support (#2039) 2024-06-10 09:22:29 +02:00
models Add Phi-3 medium support (#2039) 2024-06-10 09:22:29 +02:00
pb chore: add pre-commit (#1569) 2024-02-16 11:58:58 +01:00
utils Factor out sharding of packed tensors 2024-06-12 16:20:51 +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 ROCm and sliding windows fixes (#2033) 2024-06-10 15:09:50 +08:00
tracing.py feat(clients): Python client (#103) 2023-03-07 18:52:22 +01:00