Correct loading of models with shared tensors when using accelerator.load_state() #2875
Enabled correct loading of models with shared tensors when using acce…
b97af0f1
removed unused import
308bf249
added a test for a model with shared weights
71837647
SunMarc
approved these changes
on 2024-07-09
removed unnecessary bits
5e43b7eb
fixed linting errors
7aff18a1
muellerzr
approved these changes
on 2024-07-15
muellerzr
merged
f4f1260a
into main 2 years ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub