hf_text-generation-inference/server/text_generation_server
Daniël de Kok 85dfc39222
Add Phi-3 medium support (#2039)
Add support for Phi-3-medium

The main difference between the medium and mini models is that medium
uses grouped query attention with a packed QKV matrix. This change adds
support for GQA with packed matrixes to `Weights.get_weights_col_packed`
and uses it for Phi-3. This also allows us to remove the custom
implementation of GQA from dbrx attention loading.
2024-06-10 09:22:29 +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 Add Phi-3 medium support (#2039) 2024-06-10 09:22:29 +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