transformers
[`SwitchTransformer`] Significant performance improvement on MoE blocks
#31173
Merged
Merged
[`SwitchTransformer`] Significant performance improvement on MoE blocks #31173
ArthurZucker
merged
4 commits into
huggingface:main
from
ranggihwang:google_st_model
younesbelkada
commented
on 2024-06-03
SwitchTransformer MoE layer performance improvement
4965da6f
make fixup
8e5bb3db
younesbelkada
approved these changes
on 2024-06-03
ArthurZucker
commented
on 2024-06-03
comments about shapes
3459299e
make fixup
590cd13c
ranggihwang
changed the title
ranggihwang
changed the title
younesbelkada
approved these changes
on 2024-06-04
ArthurZucker
approved these changes
on 2024-06-06
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub