text-generation-inference/server/text_generation_server
Daniël de Kok 748764efb4 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-09-24 03:42:29 +00:00
..
layers Add Phi-3 medium support (#2039) 2024-09-24 03:42:29 +00:00
models Add Phi-3 medium support (#2039) 2024-09-24 03:42:29 +00:00
pb chore: add pre-commit (#1569) 2024-04-24 15:32:02 +03:00
utils Add Phi-3 medium support (#2039) 2024-09-24 03:42:29 +00: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-09-24 03:42:29 +00:00
interceptor.py Aligin the source code with main branch 2.0.4 2024-09-24 03:06:55 +00:00
server.py ROCm and sliding windows fixes (#2033) 2024-09-24 03:42:29 +00:00
tracing.py feat(clients): Python client (#103) 2023-03-07 18:52:22 +01:00