Update README.md
Browse files
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
|
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 = [
|