accelerate
2ea8986d
- Update tests/test_big_modeling.py
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
2 years ago
Update tests/test_big_modeling.py Co-authored-by: Marc Sun <57196510+SunMarc@users.noreply.github.com>
References
#2330 - Avoid duplicating memory for tied weights in `dispatch_model`, and in forward with offloading
Author
fxmarty
Parents
4b95a1d1
Loading