vidore
/

colqwen2-v1.0-merged

Visual Document Retrieval

Model card Files Files and versions Community

tonywu71 commited on 3 days ago

Commit

5766127

·

verified ·

1 Parent(s): 6b62b7b

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -61,7 +61,7 @@ model = ColQwen2.from_pretrained(
     "vidore/colqwen2-v1.0-merged",
     torch_dtype=torch.bfloat16,
     device_map="cuda:0",  # or "mps" if on Apple Silicon
-    attn_implementation="flash_attention_2" if is_flash_attn_2_available() else None,  # or "eager" if "mps"
 ).eval()
 processor = ColQwen2Processor.from_pretrained("vidore/colqwen2-v1.0-merged")

     "vidore/colqwen2-v1.0-merged",
     torch_dtype=torch.bfloat16,
     device_map="cuda:0",  # or "mps" if on Apple Silicon
+    attn_implementation="flash_attention_2" if is_flash_attn_2_available() else None,
 ).eval()
 processor = ColQwen2Processor.from_pretrained("vidore/colqwen2-v1.0-merged")