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 cdaa40d commit d4f8a6cCopy full SHA for d4f8a6c
src/transformers/integrations/hub_kernels.py
@@ -117,8 +117,12 @@ def use_kernel_forward_from_hub(layer_name: str):
117
},
118
"ScatterMoEGatedMLP": {
119
"cuda": {
120
- Mode.TRAINING: LayerRepository(repo_id="shawntan/scattermoe", layer_name="ScatterMoEGatedMLP"),
121
- Mode.INFERENCE: LayerRepository(repo_id="shawntan/scattermoe", layer_name="ScatterMoEGatedMLP"),
+ Mode.TRAINING: LayerRepository(
+ repo_id="kernels-community/scattermoe", layer_name="ScatterMoEGatedMLP"
122
+ ),
123
+ Mode.INFERENCE: LayerRepository(
124
125
126
127
128
"FastGELU": {
0 commit comments