[Serialization] Fix: remove context manager when loading shards + handle `mlx` weights#2709

Merged

hanouticelina merged 1 commit intomainfrom fix-loading-state-dict

Dec 13, 2024

+11-33

Commits on Dec 13, 2024

remove context manager when loading shards and handle mlx weights
hanouticelina
committed