transformers
Fix `save_pretrained` for partially offloaded models
#34890

Merged

Fix `save_pretrained` for partially offloaded models #34890

ArthurZucker merged 8 commits into huggingface:main from kylesayrs:neural-magic/patch-get_state_dict_from_offload

delete unnecessary reference

b8cc091d

kylesayrs marked this pull request as draft 1 year ago

update comment, explicit delete state_dict

041a2a5a

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

Merge branch 'main' into neural-magic/patch-get_state_dict_from_offload

8d9f484d

Rocketknight1 added bug

Rocketknight1 added Accelerate

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

Update src/transformers/modeling_utils.py

c7600493

Merge branch 'main' into neural-magic/patch-get_state_dict_from_offload

4f446740

SunMarc requested a review from

ArthurZucker 1 year ago

Merge branch 'main' into neural-magic/patch-get_state_dict_from_offload

b6e8bea9

fix style

c890858f

Merge branch 'main' into neural-magic/patch-get_state_dict_from_offload

c246f5bb

ArthurZucker approved these changes on 2024-11-28

ArthurZucker merged 4f0bf986 into main 1 year ago

Reviewers

ArthurZucker

SunMarc

muellerzr

Assignees

No one assigned

Labels

bug Accelerate

Milestone

No milestone