accelerate
State dictionary retrieval from offloaded modules
#2619

Merged

State dictionary retrieval from offloaded modules #2619

SunMarc merged 7 commits into huggingface:main from blbadger:offload-state

added get_state_dict_from_offloaded

bcfc0889

cleaned

8972167b

blbadger changed the title ~~Offload state~~ Add state dictionary retrieval from offloaded modules 1 year ago

blbadger changed the title ~~Add state dictionary retrieval from offloaded modules~~ State dictionary retrieval from offloaded modules 1 year ago

make style

dba8e1dc

muellerzr requested a review from

SunMarc 1 year ago

SunMarc commented on 2024-04-09

Update src/accelerate/utils/modeling.py

65150d6b

Merge branch 'huggingface:main' into offload-state

1eae930f

implemented suggestions, refactored, make style

4f9fd602

Merge branch 'huggingface:main' into offload-state

27101c33

SunMarc approved these changes on 2024-05-27

SunMarc merged d5d378d6 into main 1 year ago

Reviewers

SunMarc

Assignees

No one assigned

Labels

None yet

Milestone

No milestone