Remove triton mlp kernel, not compiling for some models (#37449)

* remove mlp for now * disable on docker
2025-04-11 12:47:13 +02:00
parent f797e3d98a
commit 3c39c07939
18 changed files with 2 additions and 24 deletions
--- a/src/transformers/models/aria/modeling_aria.py
+++ b/src/transformers/models/aria/modeling_aria.py
@@ -228,7 +228,6 @@ class AriaProjector(nn.Module):
        return out


-@use_kernel_forward_from_hub("MLP")
 class AriaSharedExpertsMLP(nn.Module):
    """
    Shared Expert MLP for shared experts.