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