gmastrapas
commited on
Commit
•
8ea414a
1
Parent(s):
87eae09
fix: enable flash-attn and xformers by default
Browse files- config.json +1 -1
config.json
CHANGED
@@ -36,7 +36,7 @@
|
|
36 |
"proj_bias": false,
|
37 |
"proj_type": null
|
38 |
},
|
39 |
-
"torch_dtype": "
|
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,
|