[Serialization] Fix: remove context manager when loading shards + handle mlx
weights
#7426
Job | Run time |
---|---|
42s | |
42s |
mlx
weights
#7426
Job | Run time |
---|---|
42s | |
42s |