huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-23 16:02:10 +00:00
Code Issues Packages Projects Releases Wiki Activity
1057f28128
text-generation-inference/server/text_generation_server/layers/gptq
History
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
..
__init__.py Handle GPTQ-Marlin loading in GPTQMarlinWeightLoader (#2300) 2024-09-25 05:55:39 +00:00
custom_autotune.py Some small fixes for the Torch 2.4.0 update (#2304) 2024-09-25 05:40:25 +00:00
exllama.py Fix GPTQWeight import (#2020) 2024-09-24 03:34:15 +00:00
exllamav2.py Do not initialize scratch space when there are no ExLlamaV2 layers (#2015) 2024-09-24 03:32:55 +00:00
quant_linear.py feat: add ruff and resolve issue (#2262) 2024-09-25 05:46:24 +00:00
quantize.py server quantize: store quantizer config in standard format (#2299) 2024-09-25 05:50:17 +00:00
utils.py feat: add ruff and resolve issue (#2262) 2024-09-25 05:46:24 +00:00