remove context manager when loading shards and handle mlx weights (#… #6960
| Job | Run time |
|---|---|
| 4m 43s | |
| 1m 29s | |
| 4m 0s | |
| 1m 34s | |
| 3m 26s | |
| 51s | |
| 1m 31s | |
| 55s | |
| 1m 48s | |
| 56s | |
| 1m 20s | |
| 1m 0s | |
| 23m 33s |
| Job | Run time |
|---|---|
| 4m 43s | |
| 1m 29s | |
| 4m 0s | |
| 1m 34s | |
| 3m 26s | |
| 51s | |
| 1m 31s | |
| 55s | |
| 1m 48s | |
| 56s | |
| 1m 20s | |
| 1m 0s | |
| 23m 33s |