Feat: support DTensor when saving #3042
S1ro1
changed the title tmp: return tensor.nbytes for get_torch_storage_size Feat: support DTensor when saving 232 days ago
S1ro1
marked this pull request as ready for review 232 days ago
Feat: support DTensor for storage size and id
25256775
S1ro1
force pushed
from
2768611b
to
94abfb4f
225 days ago
Feat: tests
f2f660ce
S1ro1
force pushed
from
a9589e1c
to
f2f660ce
225 days ago
Merge branch 'main' into transformers-save-dtensor
68ed3d51
Wauplin
approved these changes
on 2025-05-15
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub