accelerate
correct the return statement of _init_infer_auto_device_map
#3279

Merged

correct the return statement of _init_infer_auto_device_map #3279

muellerzr merged 1 commit into huggingface:main from Nech-C:fix_init_infer_auto

fix: add max_memory to _init_infer_auto_device_map's return statement

54603dbb

SunMarc approved these changes on 2024-12-09

SunMarc requested a review from

muellerzr 1 year ago

muellerzr approved these changes on 2024-12-13

muellerzr merged 200c9eb7 into main 1 year ago

Reviewers

SunMarc

muellerzr

Assignees

No one assigned

Labels

None yet

Milestone

No milestone