accelerate
Revert default behavior of `get_state_dict_from_offload`
#3253

Merged

Revert default behavior of `get_state_dict_from_offload` #3253

muellerzr merged 3 commits into huggingface:main from kylesayrs:neural-magic/get_state_dict_from_offload-behavior

change default to None

d1e8ddde

kylesayrs marked this pull request as draft 1 year ago

introduce move_to_device argument

44605489

remove move_to_device

dba48247

kylesayrs marked this pull request as ready for review 1 year ago

muellerzr approved these changes on 2024-11-25

muellerzr requested a review from

SunMarc 1 year ago

SunMarc approved these changes on 2024-11-25

muellerzr requested a review from

ArthurZucker 1 year ago

muellerzr merged f8c77f05 into main 1 year ago

Reviewers

SunMarc

muellerzr

ArthurZucker

Assignees

No one assigned

Labels

None yet

Milestone

No milestone