Skip to content

Commit d4f8a6c

Browse files
committed
Change kernel repo_id
1 parent cdaa40d commit d4f8a6c

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

src/transformers/integrations/hub_kernels.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -117,8 +117,12 @@ def use_kernel_forward_from_hub(layer_name: str):
117117
},
118118
"ScatterMoEGatedMLP": {
119119
"cuda": {
120-
Mode.TRAINING: LayerRepository(repo_id="shawntan/scattermoe", layer_name="ScatterMoEGatedMLP"),
121-
Mode.INFERENCE: LayerRepository(repo_id="shawntan/scattermoe", layer_name="ScatterMoEGatedMLP"),
120+
Mode.TRAINING: LayerRepository(
121+
repo_id="kernels-community/scattermoe", layer_name="ScatterMoEGatedMLP"
122+
),
123+
Mode.INFERENCE: LayerRepository(
124+
repo_id="kernels-community/scattermoe", layer_name="ScatterMoEGatedMLP"
125+
),
122126
},
123127
},
124128
"FastGELU": {

0 commit comments

Comments
 (0)