farpluto commited on
Commit
8dfcd85
1 Parent(s): 3bd06fc

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +2 -2
config.json CHANGED
@@ -17,7 +17,7 @@
17
  "architectures": [
18
  "InternLM2ForCausalLM"
19
  ],
20
- "attn_implementation": "flash_attention_2",
21
  "auto_map": {
22
  "AutoConfig": "configuration_internlm2.InternLM2Config",
23
  "AutoModel": "modeling_internlm2.InternLM2ForCausalLM",
@@ -139,6 +139,6 @@
139
  "torch_dtype": "bfloat16",
140
  "transformers_version": "4.37.2",
141
  "use_bfloat16": true,
142
- "use_flash_attn": true
143
  }
144
  }
 
17
  "architectures": [
18
  "InternLM2ForCausalLM"
19
  ],
20
+ "attn_implementation": "eager",
21
  "auto_map": {
22
  "AutoConfig": "configuration_internlm2.InternLM2Config",
23
  "AutoModel": "modeling_internlm2.InternLM2ForCausalLM",
 
139
  "torch_dtype": "bfloat16",
140
  "transformers_version": "4.37.2",
141
  "use_bfloat16": true,
142
+ "use_flash_attn": false
143
  }
144
  }