fix: key in config to match fa flag

config.json

@@ -76,7 +76,7 @@
   "torch_dtype": "bfloat16",
   "transformers_version": null,
   "use_cache": true,
-  "use_flash_attention_2": true,
+  "use_flash_attn": true,
   "use_flash_depthwise": false,
   "use_flash_rmsnorm": true,
   "use_flashfft": false,