Nicolas Patry b70ae0969f Prefix caching (#2402) ... * Prefix caching WIP * Fixing prefix attention. * Fixing flashinfer import. * Fixing black. * Fixing medusa (still wrong outputs, but functional). * Just medusa values now. * Fixing medusa without prefix caching. * Fixing prefix caching. * Medusa requires reshaping. * Removing the logs. * Remove router.nix * Fixup: - Remove logs - Disable VLMs (they do not work) - Disable prefix caching when user wants prefill logprobs. * Update flake.lock --------- Co-authored-by: Daniël de Kok <me@danieldk.eu>		2024-08-20 11:15:30 +02:00
..
attention	Prefix caching (#2402)	2024-08-20 11:15:30 +02:00
awq	feat: add ruff and resolve issue (#2262)	2024-07-26 10:29:09 -04:00
gptq	Upgrading exl2. (#2415)	2024-08-14 11:58:08 +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	Prefix caching (#2402)	2024-08-20 11:15:30 +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