Commit 57e945fd authored by zhuwenwen's avatar zhuwenwen
Browse files

add self

parent ab613ab4
...@@ -148,6 +148,7 @@ class RMSNorm(CustomOp): ...@@ -148,6 +148,7 @@ class RMSNorm(CustomOp):
@staticmethod @staticmethod
def forward_static( def forward_static(
self,
x: torch.Tensor, x: torch.Tensor,
variance_epsilon: float, variance_epsilon: float,
hidden_size: int, hidden_size: int,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment