kirp@umich.edu commited on
Commit
2e9803a
1 Parent(s): 24c6540
Files changed (1) hide show
  1. config.json +2 -2
config.json CHANGED
@@ -80,7 +80,7 @@
80
  "typical_p": 1.0,
81
  "use_bfloat16": false,
82
  "use_cache": true,
83
- "_attn_implementation": "eager",
84
  "vocab_size": 108481
85
  },
86
  "torch_dtype": "float32",
@@ -88,7 +88,7 @@
88
  "vision_config": {
89
  "_name_or_path": "",
90
  "add_cross_attention": false,
91
- "_attn_implementation": "eager",
92
  "architectures": null,
93
  "attention_dropout": 0.0,
94
  "bad_words_ids": null,
 
80
  "typical_p": 1.0,
81
  "use_bfloat16": false,
82
  "use_cache": true,
83
+ "_attn_implementation": "flash_attention_2",
84
  "vocab_size": 108481
85
  },
86
  "torch_dtype": "float32",
 
88
  "vision_config": {
89
  "_name_or_path": "",
90
  "add_cross_attention": false,
91
+ "_attn_implementation": "flash_attention_2",
92
  "architectures": null,
93
  "attention_dropout": 0.0,
94
  "bad_words_ids": null,