d3p4rt
/

newtype-cognition

Image-Text-to-Text

document-understanding

vision-language

Model card Files Files and versions

d3p4rt commited on May 16

Commit

db396b3

·

verified ·

1 Parent(s): 8fac68d

Update modeling_florence2.py

Files changed (1) hide show

modeling_florence2.py +3 -2

modeling_florence2.py CHANGED Viewed

@@ -663,8 +663,9 @@ class DaViT(nn.Module):
-if is_flash_attn_2_available():
     from flash_attn import flash_attn_func, flash_attn_varlen_func
     from flash_attn.bert_padding import index_first_axis, pad_input, unpad_input  # noqa

+# flash_attn disabled for CPU compatibility
+#if is_flash_attn_2_available():
+if False:
     from flash_attn import flash_attn_func, flash_attn_varlen_func
     from flash_attn.bert_padding import index_first_axis, pad_input, unpad_input  # noqa