Zymrael commited on
Commit
21ca99c
1 Parent(s): 7a57d19

fix: fa key in config update

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -81,7 +81,7 @@
81
  "torch_dtype": "bfloat16",
82
  "transformers_version": null,
83
  "use_cache": true,
84
- "use_flash_attention_2": true,
85
  "use_flash_depthwise": false,
86
  "use_flash_rmsnorm": false,
87
  "use_flashfft": false,
 
81
  "torch_dtype": "bfloat16",
82
  "transformers_version": null,
83
  "use_cache": true,
84
+ "use_flash_attn": true,
85
  "use_flash_depthwise": false,
86
  "use_flash_rmsnorm": false,
87
  "use_flashfft": false,