3 Commits

Author	SHA1	Message	Date
Daniël de Kok	1028996fb3	flashinfer: pass window size and dtype (#2574)	2024-09-28 18:41:41 +02:00
Nicolas Patry	dd8691b7c5	More tensor cores. (#2558) ... * More tensor cores. * Fixing the logic. * Gemma is modified by this.	2024-09-24 23:57:26 +02:00
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