Update README.md
Browse files
README.md
CHANGED
@@ -49,8 +49,8 @@ We train on an 8 GPU setup with data parallelism, a learning rate of 5e-5 with l
|
|
49 |
|
50 |
## Usage
|
51 |
|
52 |
-
Make sure `colpali-engine` is installed from source or with a version superior to 0.3.
|
53 |
-
`transformers` version must be > 4.
|
54 |
|
55 |
```bash
|
56 |
pip install git+https://github.com/illuin-tech/colpali
|
@@ -63,11 +63,11 @@ from PIL import Image
|
|
63 |
from colpali_engine.models import ColQwen2, ColQwen2Processor
|
64 |
|
65 |
model = ColQwen2.from_pretrained(
|
66 |
-
"
|
67 |
torch_dtype=torch.bfloat16,
|
68 |
device_map="cuda:0", # or "mps" if on Apple Silicon
|
69 |
).eval()
|
70 |
-
processor = ColQwen2Processor.from_pretrained("
|
71 |
|
72 |
# Your inputs
|
73 |
images = [
|
|
|
49 |
|
50 |
## Usage
|
51 |
|
52 |
+
Make sure `colpali-engine` is installed from source or with a version superior to 0.3.4.
|
53 |
+
`transformers` version must be > 4.46.1.
|
54 |
|
55 |
```bash
|
56 |
pip install git+https://github.com/illuin-tech/colpali
|
|
|
63 |
from colpali_engine.models import ColQwen2, ColQwen2Processor
|
64 |
|
65 |
model = ColQwen2.from_pretrained(
|
66 |
+
"vidore/colqwen2-v1.0",
|
67 |
torch_dtype=torch.bfloat16,
|
68 |
device_map="cuda:0", # or "mps" if on Apple Silicon
|
69 |
).eval()
|
70 |
+
processor = ColQwen2Processor.from_pretrained("vidore/colqwen2-v1.0")
|
71 |
|
72 |
# Your inputs
|
73 |
images = [
|