Nicolas Patry fb2f74e2b9 Refactor dead code - Removing all `flash_xxx.py` files. (#2166) ... * Refactor dead code. * First working step. * Remove a lot of duplicated code. * More dead code. * More cleanup. * Fix Santacoder test. * Fixing the simple tests. * Fixing sharding. * Fixes for VLM. * Fixing santacoder (num_kv_heads hardcoded). * Removing more dead code. * Fixing `config.n_head`. * Stopping earlier because of `<end_of_utterance>` in idefics2. * Addresses comments. * Removing the dead code. * Fuse back mistral into FlashCausalLM. * Finish removal. * Fixing docs + causal_lm `batch_class`. * Fixing docs + causal.lm. * Add default to Gemma Causality. * Default value for gemma/gemma2. * Wrong default.		2024-07-05 10:29:56 +02:00
..
models	Refactor dead code - Removing all `flash_xxx.py` files. (#2166)	2024-07-05 10:29:56 +02:00
utils	feat: add simple tests for weights (#2092)	2024-06-25 12:22:59 +02:00
conftest.py	feat: support typical sampling (#114)	2023-03-09 11:33:57 +01:00