vidore
/

colSmol-256M-base

vidore-experimental

Model card Files Files and versions

manu commited on Jan 23, 2025

Commit

bb5e38a

·

verified ·

1 Parent(s): e1bc0e2

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -58,7 +58,7 @@ from PIL import Image
 from colpali_engine.models import ColIdefics3, ColIdefics3Processor
 model = ColIdefics3.from_pretrained(
-        "vidore/colsmolvlm-alpha",
         torch_dtype=torch.bfloat16,
         device_map="cuda:0",
         attn_implementation="flash_attention_2" # or eager

 from colpali_engine.models import ColIdefics3, ColIdefics3Processor
 model = ColIdefics3.from_pretrained(
+        "vidore/colSmol-256M-base",
         torch_dtype=torch.bfloat16,
         device_map="cuda:0",
         attn_implementation="flash_attention_2" # or eager