4 Commits

Author	SHA1	Message	Date
Alex Barron	d4ef909d4a	Length masking for batch inputs (#1173) ... * length masking * add mask to mlx_lm model interface * remove lengths * fix test: * comment + fix	2024-12-18 19:43:52 -08:00
Alex Barron	85ffd2c96a	Quantized KV Cache (#1075) ... * add QuantizedKVCache * simplify * add tests * single sdpa function * fix sed * in place * fix tests * support different k and v head dims	2024-10-31 16:59:52 -07:00
Awni Hannun	fca087be49	More cache improvements (#1015) ... * fix rotating kv cache for chat use case * reorg + fixes to caching, unify prompt caching across types and use cases for e.g. caching during a chat * nit in chat * fix tests * fix tests * fix tests * docs * chat command * comments + docs * Define meta_state on all Cache implementations * fixes + trim_prompt_cache api * fix default model --------- Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>	2024-10-07 20:45:51 -07:00
L	fc93c55723	feat(mlx_lm): Nemotron (#949) ... * feat: Nemotron https://huggingface.co/nvidia/Minitron-4B-Base This is basically Llama with partial RoPE and LayerNorm instead of BatchNorm. Also they add 1 to the LayerNorm weight for some reason. * fixup! feat: Nemotron * nits --------- Co-authored-by: Awni Hannun <awni@apple.com>	2024-08-29 21:08:57 -07:00