11 Commits

Author	SHA1	Message	Date
Awni Hannun	78c431dc25	cleanup whisper a little (#639)	2024-03-30 13:13:58 -07:00
Awni Hannun	ec14583c2a	work with tuple shape (#393)	2024-02-01 13:03:47 -08:00
Awni Hannun	80d18671ad	[Lora] Fix generate (#282) ... * fix generate * update readme, fix test, better default * nits * typo	2024-01-10 16:13:06 -08:00
bofeng huang	bf9926489e	[Whisper] Add word timestamps and confidence scores (#201) ... * Add word timestamps and confidence scores * Create a separate forward_with_cross_qk function * Move multiple ops from np to mlx, clean comments * Save alignment_heads * Cast qk to fp32 * Add test for word-level timestamps and confidence scores * format + readme * nit --------- Co-authored-by: Awni Hannun <awni@apple.com>	2024-01-07 10:01:29 -08:00
Awni Hannun	a5d6d0436c	Support Hugging Face models (#215) ... * support hf direct models	2024-01-03 15:13:26 -08:00
bofeng huang	581a5733a1	[Whisper] Load customized MLX model & Quantization (#191) ... * Add option to load customized mlx model * Add quantization * Apply reviews * Separate model conversion and loading * Update test * Fix benchmark * Add notes about conversion * Improve doc	2023-12-29 10:22:15 -08:00
Awni Hannun	27c0a8c002	Add llms subdir + update README (#145) ... * add llms subdir + update README * nits * use same pre-commit as mlx * update readmes a bit * format	2023-12-20 10:22:25 -08:00
Awni Hannun	b863e7cca0	format	2023-12-14 16:56:50 -08:00
Awni Hannun	6e723a015a	whisper default in fp16	2023-12-12 07:37:35 -08:00
Awni Hannun	31bc57c4ff	add copyright in source	2023-11-30 11:08:53 -08:00
Awni Hannun	b243c1d8f4	a few examples	2023-11-29 08:17:26 -08:00