mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-08-29 10:37:15 +08:00
add coauthor :)
Co-authored-by: n8programs <43304488+N8python@users.noreply.github.com>
This commit is contained in:
parent
d7d70487eb
commit
406c7f300f
@ -29,7 +29,6 @@ class ModelArgs(BaseModelArgs):
|
|||||||
|
|
||||||
|
|
||||||
class LayerNorm2D(nn.Module):
|
class LayerNorm2D(nn.Module):
|
||||||
|
|
||||||
def __init__(self, d1, d2, eps):
|
def __init__(self, d1, d2, eps):
|
||||||
super().__init__()
|
super().__init__()
|
||||||
self.weight = mx.zeros((d1, d2))
|
self.weight = mx.zeros((d1, d2))
|
||||||
|
Loading…
Reference in New Issue
Block a user