Update README.md
Browse files
README.md
CHANGED
@@ -58,7 +58,7 @@ from PIL import Image
|
|
58 |
from colpali_engine.models import ColIdefics3, ColIdefics3Processor
|
59 |
|
60 |
model = ColIdefics3.from_pretrained(
|
61 |
-
"vidore/
|
62 |
torch_dtype=torch.bfloat16,
|
63 |
device_map="cuda:0",
|
64 |
attn_implementation="flash_attention_2" # or eager
|
|
|
58 |
from colpali_engine.models import ColIdefics3, ColIdefics3Processor
|
59 |
|
60 |
model = ColIdefics3.from_pretrained(
|
61 |
+
"vidore/colSmol-500M-base",
|
62 |
torch_dtype=torch.bfloat16,
|
63 |
device_map="cuda:0",
|
64 |
attn_implementation="flash_attention_2" # or eager
|