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

Commit
2 years ago
Correct loading of models with shared tensors when using accelerator.load_state() (#2875) * Enabled correct loading of models with shared tensors when using accelerator.load_state() * removed unused import * added a test for a model with shared weights * removed unnecessary bits * fixed linting errors
Author
Parents
Loading