mlx/python
Anchen f6feb61f92
feat: add support for saving safetensors in the save_weights (#497)
* feat: add save safetensors support in module save_weights

* chore: checking missing changes

* Update python/mlx/nn/layers/base.py

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>

* chore: update docstring for load_weights

---------

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
2024-01-19 06:19:33 -08:00
..
mlx feat: add support for saving safetensors in the save_weights (#497) 2024-01-19 06:19:33 -08:00
src Support disable metal buffer cache to prevent performance degradation caused by large memory caching (#390) 2024-01-18 08:33:34 -08:00
tests feat: add support for saving safetensors in the save_weights (#497) 2024-01-19 06:19:33 -08:00