transformers
Adding _tie_weights() to prediction heads to support low_cpu_mem_usage=True
#29024

Merged

Adding _tie_weights() to prediction heads to support low_cpu_mem_usage=True #29024

ArthurZucker merged 5 commits into huggingface:main from hackyon:fix-more-tied-weights

hackyon force pushed 1 year ago

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

hackyon force pushed 1 year ago

amyeroberts commented on 2024-02-15

ArthurZucker commented on 2024-02-19

hackyon force pushed 1 year ago

hackyon changed the title ~~Adding _tie_weights() to more models~~ Adding _tie_weights() to prediction heads to support low_cpu_mem_usage=True 1 year ago

hackyon force pushed 1 year ago

SunMarc commented on 2024-02-27

hackyon force pushed 1 year ago

SunMarc approved these changes on 2024-03-26

Adding _tie_weights() to prediction heads to support low_cpu_mem_usag…

1184601d

Testing for the non-safe-tensors case, since the default is safe-tens…

3dbcfa11

Running fixup/fix-copies

e55d4a81

Adding accelerate annotations to tests

52695f46

hackyon force pushed to 52695f46 1 year ago

SunMarc requested a review from

ArthurZucker 1 year ago

Merge remote-tracking branch 'upstream/main' into fix-more-tied-weights

3d5a7b75

ArthurZucker approved these changes on 2024-05-07

ArthurZucker merged 54a2361a into main 1 year ago

hackyon deleted the fix-more-tied-weights branch 1 year ago

Reviewers

ArthurZucker

SunMarc

amyeroberts

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

transformers Adding _tie_weights() to prediction heads to support low_cpu_mem_usage=True #29024 Merged

Adding _tie_weights() to prediction heads to support low_cpu_mem_usage=True #29024

transformers
Adding _tie_weights() to prediction heads to support low_cpu_mem_usage=True
#29024

Merged