Skip to content

Commit 0a7afad

Browse files
Update model.py
1 parent 01062c9 commit 0a7afad

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

‎gemma/model.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -194,7 +194,7 @@ def __init__(
194194

195195
def forward(self, x):
196196
gate = self.gate_proj(x)
197-
gate = F.gelu(gate)
197+
gate = F.gelu(gate, approximate="tanh")
198198
up = self.up_proj(x)
199199
fuse = gate * up
200200
outputs = self.down_proj(fuse)

0 commit comments

Comments
 (0)