mx262 commited on
Commit
786ef79
1 Parent(s): 59a5edb

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -17,7 +17,7 @@
17
  "architectures": [
18
  "InternLM2ForCausalLM"
19
  ],
20
- "attn_implementation": "eager",
21
  "auto_map": {
22
  "AutoConfig": "configuration_internlm2.InternLM2Config",
23
  "AutoModel": "modeling_internlm2.InternLM2ForCausalLM",
 
17
  "architectures": [
18
  "InternLM2ForCausalLM"
19
  ],
20
+ "attn_implementation": "flash_attention_2",
21
  "auto_map": {
22
  "AutoConfig": "configuration_internlm2.InternLM2Config",
23
  "AutoModel": "modeling_internlm2.InternLM2ForCausalLM",