transformers
[CodeGen] support device_map="auto" for sharded checkpoints
#17871

Merged

[CodeGen] support device_map="auto" for sharded checkpoints #17871

patil-suraj merged 1 commit into huggingface:main from patil-suraj:fix-codegen-sharded

add _no_split_modules

0369aabf

patil-suraj requested a review from

patrickvonplaten 3 years ago

patil-suraj requested a review from

sgugger 3 years ago

sgugger approved these changes on 2022-06-24

patil-suraj merged 061a73d1 into main 3 years ago

patil-suraj deleted the fix-codegen-sharded branch 3 years ago

Reviewers

sgugger

patrickvonplaten

Assignees

No one assigned

Labels

None yet

Milestone

No milestone