accelerate
Correct loading of models with shared tensors when using accelerator.load_state()
#2875
Merged

Correct loading of models with shared tensors when using accelerator.load_state() #2875

jkuntzer
jkuntzer Enabled correct loading of models with shared tensors when using acce…
b97af0f1
HuggingFaceDocBuilderDev
jkuntzer removed unused import
308bf249
muellerzr
muellerzr commented on 2024-07-03
SunMarc
SunMarc commented on 2024-07-03
jkuntzer
jkuntzer added a test for a model with shared weights
71837647
jkuntzer
SunMarc
SunMarc approved these changes on 2024-07-09
jkuntzer
SunMarc
SunMarc commented on 2024-07-09
SunMarc
jkuntzer removed unnecessary bits
5e43b7eb
jkuntzer
SunMarc
SunMarc SunMarc requested a review from muellerzr muellerzr 2 years ago
jkuntzer fixed linting errors
7aff18a1
muellerzr
muellerzr approved these changes on 2024-07-15
muellerzr muellerzr merged f4f1260a into main 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone