[Serialization] Fix: remove context manager when loading shards + handle `mlx` weights #2709
remove context manager when loading shards and handle mlx weights
ed113dcd
Wauplin
approved these changes
on 2024-12-13
hanouticelina
deleted the fix-loading-state-dict branch 1 year ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub