Upload modeling_opt.py
Browse files- 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=
|
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
|