transformers
Make `DtensorShardOperation` work on torch 2.6 and 2.7
#49657

Merged

Make `DtensorShardOperation` work on torch 2.6 and 2.7 #49657

vasqu merged 2 commits into huggingface:main from hazelxue:fix/sharding-utils-torch26-27

Make DtensorShardOperation work on torch 2.6 and 2.7

6d61c6ef

3outeille requested a review from

3outeille 8 days ago

3outeille commented on 2026-10-16

3outeille requested a review from

ArthurZucker 8 days ago

3outeille requested a review from

vasqu 8 days ago

Resolve the Shard helper in one function instead of patching Shard

e90e0747

3outeille approved these changes on 2026-10-16

ArthurZucker approved these changes on 2026-10-16

vasqu merged 7bf3c5ec into main 8 days ago

vasqu added for patch

Reviewers

ArthurZucker

3outeille

vasqu

Assignees

No one assigned

Labels

for patch

Milestone

No milestone