transformers
423f2a31 - fix(models): Fix dtype mismatch in SwitchTransformers and TimmWrapperModel (#45074)

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
Parents
Loading