ColPali
Safetensors
English
vidore-experimental
vidore
manu commited on
Commit
d6601e2
1 Parent(s): 339abdc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
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.1.
53
- `transformers` version must be > 4.45.0.
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
- "manu/colqwen2-v1.0-alpha",
67
  torch_dtype=torch.bfloat16,
68
  device_map="cuda:0", # or "mps" if on Apple Silicon
69
  ).eval()
70
- processor = ColQwen2Processor.from_pretrained("manu/colqwen2-v1.0-alpha")
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 = [