gmastrapas commited on
Commit
8ea414a
1 Parent(s): 87eae09

fix: enable flash-attn and xformers by default

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -36,7 +36,7 @@
36
  "proj_bias": false,
37
  "proj_type": null
38
  },
39
- "torch_dtype": "bfloat16",
40
  "truncate_dim": null,
41
  "use_text_flash_attn": null,
42
  "use_vision_xformers": null,
 
36
  "proj_bias": false,
37
  "proj_type": null
38
  },
39
+ "torch_dtype": "float16",
40
  "truncate_dim": null,
41
  "use_text_flash_attn": null,
42
  "use_vision_xformers": null,