zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-08-29 16:58:45 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

revert to act_fn to silu

Browse Source
This commit is contained in:
Prince Canuma 2024-12-14 16:22:00 +01:00
parent d7c64c621a
commit 5d8b36ce7c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
llms/mlx_lm/models/cohere2.py
View File

@ -115,7 +115,7 @@ class MLP(nn.Module):
self.down_proj = nn.Linear(hidden_dim, dim, bias=False) self.down_proj = nn.Linear(hidden_dim, dim, bias=False)
def __call__(self, x): def __call__(self, x):
return self.down_proj(nn.gelu(self.gate_proj(x)) * self.up_proj(x)) return self.down_proj(nn.silu(self.gate_proj(x)) * self.up_proj(x))
class TransformerBlock(nn.Module): class TransformerBlock(nn.Module):