Skip to content

Commit 2d0ca7e

Browse files
committed
Change kernel repo_id
1 parent 98bf8c4 commit 2d0ca7e

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
@@ -88,8 +88,12 @@
8888
},
8989
"ScatterMoEGatedMLP": {
9090
"cuda": {
91-
Mode.TRAINING: LayerRepository(repo_id="shawntan/scattermoe", layer_name="ScatterMoEGatedMLP"),
92-
Mode.INFERENCE: LayerRepository(repo_id="shawntan/scattermoe", layer_name="ScatterMoEGatedMLP"),
91+
Mode.TRAINING: LayerRepository(
92+
repo_id="kernels-community/scattermoe", layer_name="ScatterMoEGatedMLP"
93+
),
94+
Mode.INFERENCE: LayerRepository(
95+
repo_id="kernels-community/scattermoe", layer_name="ScatterMoEGatedMLP"
96+
),
9397
},
9498
},
9599
"FastGELU": {

0 commit comments

Comments
 (0)