transformers
11c49ed2
- Load the state dict on CPU to prevent unnecessary GPU memory surge (#20920)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
3 years ago
Load the state dict on CPU to prevent unnecessary GPU memory surge (#20920) load the state dict on cpu.
References
#20920 - Load the state dict on CPU to prevent unnecessary GPU memory surge
#27720 - Add common processor tests
#29969 - [SigLIP] Add fast tokenizer
#32831 - [Docs] Update resources
#33111 - [Backbone] Remove out_features everywhere
#33174 - [Zero-shot image classification pipeline] Remove tokenizer_kwargs
#39821 - Support MetaCLIP 2
#59 - Fix attention mask handling in EoMT-DINOv3 converter
#62 - Add initial DEIMv2 model implementation
#65 - Fix RTDetrV2 sine position embedding ordering
Author
HarshTrivedi
Parents
0b686a8a
Loading