add coauthor :)

Co-authored-by: n8programs <43304488+N8python@users.noreply.github.com>
This commit is contained in:
Prince Canuma 2024-12-14 17:12:49 +01:00
parent d7d70487eb
commit 406c7f300f

View File

@ -29,7 +29,6 @@ class ModelArgs(BaseModelArgs):
class LayerNorm2D(nn.Module):
def __init__(self, d1, d2, eps):
super().__init__()
self.weight = mx.zeros((d1, d2))