jinaai
/

jina-clip-v2

gmastrapas commited on 6 days ago

Commit

ced12ef

•

1 Parent(s): 8ea414a

fix: enable flash-attn and xformers by default

Files changed (1) hide show

config.json CHANGED Viewed

@@ -36,7 +36,7 @@
         "proj_bias": false,
         "proj_type": null
     },
-    "torch_dtype": "float16",
     "truncate_dim": null,
     "use_text_flash_attn": null,
     "use_vision_xformers": null,

         "proj_bias": false,
         "proj_type": null
     },
+    "torch_dtype": "bfloat16",
     "truncate_dim": null,
     "use_text_flash_attn": null,
     "use_vision_xformers": null,