Skip to content

Commit c748111

Browse files
authored
Fix the MPS RMSNorm hub kernel repo id (#50714)
Fix the MPS RMSNorm hub kernel repo id (mlx-rmsnorm)
1 parent 41f9d39 commit c748111

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

‎src/transformers/integrations/hub_kernels.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -491,7 +491,7 @@ def _build_kernel_mapping() -> dict:
491491
},
492492
"mps": {
493493
Mode.INFERENCE: LayerRepository(
494-
repo_id="kernels-community/mlx_rmsnorm",
494+
repo_id="kernels-community/mlx-rmsnorm",
495495
layer_name="RMSNorm",
496496
version=1,
497497
)

0 commit comments

Comments
 (0)