ColPali
Safetensors
English
vidore
manu commited on
Commit
b8355f7
1 Parent(s): bac167c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -50,14 +50,14 @@ We train on an 8 GPU setup with data parallelism, a learning rate of 5e-5 with l
50
  import torch
51
  from PIL import Image
52
 
53
- from colpali_engine.models import Colwen2, ColQwen2Processor
54
 
55
  model = ColQwen2.from_pretrained(
56
  "manu/colqwen2-beta",
57
  torch_dtype=torch.bfloat16,
58
  device_map="cuda:0", # or "mps" if on Apple Silicon
59
  )
60
- processor = ColQwen2Processor.from_pretrained("manu/colqwen2-beta"))
61
 
62
  # Your inputs
63
  images = [
 
50
  import torch
51
  from PIL import Image
52
 
53
+ from colpali_engine.models import ColQwen2, ColQwen2Processor
54
 
55
  model = ColQwen2.from_pretrained(
56
  "manu/colqwen2-beta",
57
  torch_dtype=torch.bfloat16,
58
  device_map="cuda:0", # or "mps" if on Apple Silicon
59
  )
60
+ processor = ColQwen2Processor.from_pretrained("manu/colqwen2-beta")
61
 
62
  # Your inputs
63
  images = [