transformers
423f2a31
- fix(models): Fix dtype mismatch in SwitchTransformers and TimmWrapperModel (#45074)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
4 days ago
fix(models): Fix dtype mismatch in SwitchTransformers and TimmWrapperModel (#45074) * fix: Cast inputs to match weight dtype * new: Add test * change: Upcast to float32 instead of downcasting
References
#45074 - fix(models): Fix dtype mismatch in SwitchTransformers and TimmWrapperModel
Author
harshaljanjani
Parents
ade7a05a
Loading