Nicolas Patry a313355d2b Tied embeddings in MLP speculator. (#2473) ... * Tied embeddings in MLP speculator. * Fixing the scale_weight when users decide to not use the speculation as much as defined in the config. * Adding scaling support + optimize some ops.		2024-09-25 06:13:11 +00:00
..
attention	Lots of improvements (Still 2 allocators) (#2449)	2024-09-25 06:13:11 +00:00
awq	feat: add ruff and resolve issue (#2262)	2024-09-25 05:46:24 +00:00
gptq	fix: include create_exllama_buffers and set_device for exllama (#2407)	2024-09-25 06:06:17 +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	Prefix caching (#2402)	2024-09-25 06:10:59 +00:00
mlp.py	Tied embeddings in MLP speculator. (#2473)	2024-09-25 06:13:11 +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