Update modeling_neollm.py
Browse files- modeling_neollm.py +1 -1
modeling_neollm.py
CHANGED
|
@@ -2353,7 +2353,7 @@ class NeoLLMMLP(nn.Module):
|
|
| 2353 |
config.intermediate_size, config.hidden_size,
|
| 2354 |
bias=False, use_row_multiplier=True, use_column_multiplier=True,
|
| 2355 |
)
|
| 2356 |
-
self.act_fn = PolyNorm(exclusive_init=0.
|
| 2357 |
self.dropout = nn.Dropout(config.dropout_rate)
|
| 2358 |
|
| 2359 |
def forward(
|
|
|
|
| 2353 |
config.intermediate_size, config.hidden_size,
|
| 2354 |
bias=False, use_row_multiplier=True, use_column_multiplier=True,
|
| 2355 |
)
|
| 2356 |
+
self.act_fn = PolyNorm(exclusive_init=0.15)
|
| 2357 |
self.dropout = nn.Dropout(config.dropout_rate)
|
| 2358 |
|
| 2359 |
def forward(
|