transformers
ea41e18c
- improve from_pretrained for zero3 multi gpus mode (#24964)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
2 years ago
improve from_pretrained for zero3 multi gpus mode (#24964) * improve from_pretrained for zero3 multi gpus mode * Add check if torch.distributed.is_initialized * Revert torch.distributed --------- Co-authored-by: Stas Bekman <stas@stason.org>
References
#24964 - improve from_pretrained for zero3 multi gpus mode
Author
1ytic
Parents
95f96b45
Loading