hf_text-generation-inference/server/text_generation_server/layers
Nicolas Patry 84bc3d7b7d
Fixing import exl2 (#2399)
2024-08-12 14:08:59 +02:00
..
attention Using an enum for flash backens (paged/flashdecoding/flashinfer) (#2385) 2024-08-09 16:41:17 +02:00
awq feat: add ruff and resolve issue (#2262) 2024-07-26 10:29:09 -04:00
gptq Fixing import exl2 (#2399) 2024-08-12 14:08:59 +02:00
marlin Handle GPTQ-Marlin loading in `GPTQMarlinWeightLoader` (#2300) 2024-07-31 13:08:41 +02:00
__init__.py feat: add ruff and resolve issue (#2262) 2024-07-26 10:29:09 -04:00
bnb.py feat: add ruff and resolve issue (#2262) 2024-07-26 10:29:09 -04:00
conv.py Refactor layers. (#1866) 2024-05-13 12:44:30 +02:00
eetq.py feat(fp8): use fbgemm kernels and load fp8 weights directly (#2248) 2024-07-20 19:02:04 +02:00
exl2.py Add support for Deepseek V2 (#2224) 2024-07-19 17:23:20 +02:00
fp8.py feat: add ruff and resolve issue (#2262) 2024-07-26 10:29:09 -04:00
layernorm.py Removing IPEX_AVAIL. (#2115) 2024-06-25 13:20:57 +02:00
linear.py feat: add ruff and resolve issue (#2262) 2024-07-26 10:29:09 -04:00
lora.py feat: add ruff and resolve issue (#2262) 2024-07-26 10:29:09 -04:00
medusa.py fix: use path inside of speculator config (#1935) 2024-05-22 20:46:29 +02:00
mlp.py MLPSpeculator. (#1865) 2024-05-14 12:33:18 +02:00
rotary.py feat: add ruff and resolve issue (#2262) 2024-07-26 10:29:09 -04:00
speculative.py feat: add ruff and resolve issue (#2262) 2024-07-26 10:29:09 -04:00
tensor_parallel.py feat: add ruff and resolve issue (#2262) 2024-07-26 10:29:09 -04:00