Skip to content

Commit a6dcc2c

Browse files
authored
Update NPU kernel repository and layer name
1 parent 266d3b0 commit a6dcc2c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/transformers/integrations/hub_kernels.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -86,8 +86,8 @@ def use_kernel_forward_from_hub(layer_name: str):
8686
},
8787
"npu": {
8888
Mode.INFERENCE: LayerRepository(
89-
repo_id="kernels-community/liger_kernels",
90-
layer_name="LigerRMSNorm",
89+
repo_id="kernels-ext-npu/rmsnorm",
90+
layer_name="rmsnorm",
9191
)
9292
},
9393
},

0 commit comments

Comments
 (0)