robinzixuan commited on
Commit
df61511
·
verified ·
1 Parent(s): fd77078

Upload modeling_opt.py

Browse files
Files changed (1) hide show
  1. modeling_opt.py +1 -1
modeling_opt.py CHANGED
@@ -184,7 +184,7 @@ class OPTAttention(nn.Module):
184
  attn_gate_mlp2=False,
185
  attn_gate_linear_all_features=False,
186
  fine_tuning=False,
187
- attn_softmax=softmax,
188
  ):
189
  super().__init__()
190
  self.embed_dim = embed_dim
 
184
  attn_gate_mlp2=False,
185
  attn_gate_linear_all_features=False,
186
  fine_tuning=False,
187
+ attn_softmax='vanilla',
188
  ):
189
  super().__init__()
190
  self.embed_dim = embed_dim