accelerate
Fix error when max_memory argument is in unexpected order
#1759

Merged

Fix error when max_memory argument is in unexpected order #1759

sgugger merged 8 commits into huggingface:main from ranchlai:max_memory

sort the user-provided max_memory keys in gpu-cpu-disk order

f84ac36b

fixed the bug by adding disk to main devices

394d937f

ranchlai changed the title ~~Max memory~~ Fix error when max_memory argument is in unexpected order 3 years ago

add checking for max_memory argument

54989c29

ranchlai force pushed from 301c034a to 54989c29 3 years ago

muellerzr requested a review from

sgugger 3 years ago

sgugger commented on 2023-07-24

Update src/accelerate/utils/modeling.py

1c4b6920

Update src/accelerate/utils/modeling.py

9b14ed4b

Update src/accelerate/utils/modeling.py

98a9dd6b

fix typo

630c82be

fix typos

cbcbe240

sgugger approved these changes on 2023-07-24

sgugger merged f3b7f9cf into main 3 years ago

Reviewers

sgugger

Assignees

No one assigned

Labels

None yet

Milestone

No milestone