KitsuVp commited on
Commit
55f56cb
·
verified ·
1 Parent(s): fe8dd15

Update modeling_neollm.py

Browse files
Files changed (1) hide show
  1. 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.05)
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(