transformers
fix(moe): Handle dtype mismatch in torch._grouped_mm with autocast
#43839

Merged

fix(moe): Handle dtype mismatch in torch._grouped_mm with autocast #43839

ArthurZucker merged 2 commits into huggingface:main from fix/phimoe-autocast-dtype

fix(moe): Handle dtype mismatch in grouped_mm with autocast

1d3bb2d1

vasqu requested a review from

IlyasMoutawwakil 197 days ago

refactor: use autocast(enabled=False) instead of manual dtype cast

60b7a73b

IlyasMoutawwakil approved these changes on 2026-02-11

ArthurZucker commented on 2026-02-11

ArthurZucker merged 007bb8c3 into main 195 days ago

Reviewers

IlyasMoutawwakil

ArthurZucker

Assignees

No one assigned

Labels

None yet

Milestone

No milestone