transformers
fix(models): Fix dtype mismatch in SwitchTransformers and TimmWrapperModel
#45074
Merged
Overview
Commits
4
Changes
View On GitHub
Commits
  • fix: Cast inputs to match weight dtype
    harshaljanjani committed 227 days ago
  • new: Add test
    harshaljanjani committed 227 days ago
  • Merge branch 'main' into fix/switch-transformers-timm-wrapper-bf16-dtype
    harshaljanjani committed 224 days ago
  • change: Upcast to float32 instead of downcasting
    harshaljanjani committed 224 days ago
FAQ Terms Privacy Refunds Impressum
Loading