transformers
442d356a - Add moe kernels (#37376)

Commit
247 days ago
Add moe kernels (#37376) * the fix that did not get in * add kernels * full graph does not work * simpler is better * Update src/transformers/integrations/hub_kernels.py Co-authored-by: Daniël de Kok <me@danieldk.eu> * Update src/transformers/integrations/fbgemm_fp8.py Co-authored-by: Daniël de Kok <me@danieldk.eu> * Update src/transformers/integrations/hub_kernels.py Co-authored-by: Daniël de Kok <me@danieldk.eu> * fixup --------- Co-authored-by: Daniël de Kok <me@danieldk.eu>
Author
Parents
Loading