Update README.md
Browse files
README.md
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
---
|
2 |
-
base_model:
|
3 |
library_name: peft
|
4 |
tags:
|
5 |
- vidore-experimental
|
@@ -56,12 +56,12 @@ from PIL import Image
|
|
56 |
from colpali_engine.models import ColIdefics3, ColIdefics3Processor
|
57 |
|
58 |
model = ColIdefics3.from_pretrained(
|
59 |
-
"vidore/
|
60 |
torch_dtype=torch.bfloat16,
|
61 |
device_map="cuda:0",
|
62 |
attn_implementation="flash_attention_2" # or eager
|
63 |
).eval()
|
64 |
-
processor = ColIdefics3Processor.from_pretrained("
|
65 |
|
66 |
# Your inputs
|
67 |
images = [
|
|
|
1 |
---
|
2 |
+
base_model: vidore/ColSmolVLM-base
|
3 |
library_name: peft
|
4 |
tags:
|
5 |
- vidore-experimental
|
|
|
56 |
from colpali_engine.models import ColIdefics3, ColIdefics3Processor
|
57 |
|
58 |
model = ColIdefics3.from_pretrained(
|
59 |
+
"vidore/colsmolvlm-alpha",
|
60 |
torch_dtype=torch.bfloat16,
|
61 |
device_map="cuda:0",
|
62 |
attn_implementation="flash_attention_2" # or eager
|
63 |
).eval()
|
64 |
+
processor = ColIdefics3Processor.from_pretrained("vidore/colsmolvlm-alpha")
|
65 |
|
66 |
# Your inputs
|
67 |
images = [
|