transformers
84a67891
- Enable different torch dtype in sub models (#34873)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
Enable different torch dtype in sub models (#34873) * fix * fix test * add tests * add more tests * fix tests * supposed to be a torch.dtype test * handle BC and make fp32 default
References
#34873 - Enable different torch dtype in sub models
Author
zucchini-nlp
Parents
87089176
Loading