Update README.md

README.md

@@ -65,11 +65,11 @@ from transformers.utils.import_utils import is_flash_attn_2_available
 from colpali_engine.models import ColQwen2, ColQwen2Processor
 
 model = ColQwen2.from_pretrained(
-        "vidore/colqwen2-v1.0",
-        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()
+    "vidore/colqwen2-v1.0",
+    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")
 
 # Your inputs