text-generation-inference/server/text_generation_server/layers
Daniël de Kok 468e5c6874 Handle GPTQ-Marlin loading in GPTQMarlinWeightLoader (#2300)
The `GPTWeightLoader` was structured like this in pseudocode:

if marlin:
  Set up tensors in a way that GPTQ-Marlin expects
else:
  Set up tensors in a way that ExLlama/GPTQ/AWQ expect

However, the GPT-Marlin implementation details should really be in the
`marlin` module. So move the former part out to a separate
`GPTQMarlinWeightsLoader`.
2024-09-25 05:55:39 +00:00
..
attention feat: add ruff and resolve issue (#2262) 2024-09-25 05:46:24 +00:00
awq feat: add ruff and resolve issue (#2262) 2024-09-25 05:46:24 +00:00
gptq Handle GPTQ-Marlin loading in GPTQMarlinWeightLoader (#2300) 2024-09-25 05:55:39 +00:00
marlin Handle GPTQ-Marlin loading in GPTQMarlinWeightLoader (#2300) 2024-09-25 05:55:39 +00:00
__init__.py feat: add ruff and resolve issue (#2262) 2024-09-25 05:46:24 +00:00
bnb.py feat: add ruff and resolve issue (#2262) 2024-09-25 05:46:24 +00:00
conv.py Refactor layers. (#1866) 2024-07-17 05:36:58 +00:00
eetq.py feat(fp8): use fbgemm kernels and load fp8 weights directly (#2248) 2024-09-25 05:30:41 +00:00
exl2.py Add support for Deepseek V2 (#2224) 2024-09-25 05:27:40 +00:00
fp8.py feat: add ruff and resolve issue (#2262) 2024-09-25 05:46:24 +00:00
layernorm.py Removing IPEX_AVAIL. (#2115) 2024-09-24 03:52:23 +00:00
linear.py feat: add ruff and resolve issue (#2262) 2024-09-25 05:46:24 +00:00
lora.py feat: add ruff and resolve issue (#2262) 2024-09-25 05:46:24 +00:00
medusa.py fix: use path inside of speculator config (#1935) 2024-07-17 05:36:58 +00:00
mlp.py MLPSpeculator. (#1865) 2024-07-17 05:36:58 +00:00
rotary.py feat: add ruff and resolve issue (#2262) 2024-09-25 05:46:24 +00:00
speculative.py feat: add ruff and resolve issue (#2262) 2024-09-25 05:46:24 +00:00
tensor_parallel.py feat: add ruff and resolve issue (#2262) 2024-09-25 05:46:24 +00:00