Chime Ogbuji f6283ef7ce Configurable LR schedulers (#604) ... * Initial config handler and test * Added means to run from CLI * Update lora config loading and tests * Constrain scheduler config (warmup and minimum LR) for each kind * Update reference to moved schedule_config module * Minor fix * Fix typos * Moved build_schedule and tests * nits in schedule config * flake * fix path --------- Co-authored-by: Awni Hannun <awni@apple.com>		2024-03-29 13:41:10 -07:00
..
__init__.py	feat: move lora into mlx-lm (#337)	2024-01-23 08:44:37 -08:00
datasets.py	Support for OpenAI’s fine-tuning dataset format (#548)	2024-03-19 16:45:46 -07:00
lora.py	cast around lora adapters (#613)	2024-03-24 19:34:51 -07:00
trainer.py	chore: fix loar for moe model (#608)	2024-03-23 07:22:11 -07:00
utils.py	Configurable LR schedulers (#604)	2024-03-29 13:41:10 -07:00