We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 98bf8c4 commit 2d0ca7eCopy full SHA for 2d0ca7e
src/transformers/integrations/hub_kernels.py
@@ -88,8 +88,12 @@
88
},
89
"ScatterMoEGatedMLP": {
90
"cuda": {
91
- Mode.TRAINING: LayerRepository(repo_id="shawntan/scattermoe", layer_name="ScatterMoEGatedMLP"),
92
- Mode.INFERENCE: LayerRepository(repo_id="shawntan/scattermoe", layer_name="ScatterMoEGatedMLP"),
+ Mode.TRAINING: LayerRepository(
+ repo_id="kernels-community/scattermoe", layer_name="ScatterMoEGatedMLP"
93
+ ),
94
+ Mode.INFERENCE: LayerRepository(
95
96
97
98
99
"FastGELU": {
0 commit comments