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
..
benches	Keeping the benchmark somewhere (#2401)	2024-08-12 15:22:02 +02:00
src	Prefix caching (#2402)	2024-08-20 11:15:30 +02:00
Cargo.toml	Keeping the benchmark somewhere (#2401)	2024-08-12 15:22:02 +02:00
build.rs	Rebase TRT-llm (#2331)	2024-07-31 10:33:10 +02:00