zR
commited on
Commit
•
a7cafb0
1
Parent(s):
534e83c
add flash-attn
Browse files- config.json +1 -0
config.json
CHANGED
@@ -17,6 +17,7 @@
|
|
17 |
"apply_residual_connection_post_layernorm": false,
|
18 |
"attention_dropout": 0.0,
|
19 |
"attention_softmax_in_fp32": true,
|
|
|
20 |
"bias_dropout_fusion": true,
|
21 |
"ffn_hidden_size": 13696,
|
22 |
"fp32_residual_connection": false,
|
|
|
17 |
"apply_residual_connection_post_layernorm": false,
|
18 |
"attention_dropout": 0.0,
|
19 |
"attention_softmax_in_fp32": true,
|
20 |
+
"attn_implementation": "sdpa",
|
21 |
"bias_dropout_fusion": true,
|
22 |
"ffn_hidden_size": 13696,
|
23 |
"fp32_residual_connection": false,
|