Update README.md
Browse files
README.md
CHANGED
@@ -63,7 +63,7 @@ model = ColQwen2.from_pretrained(
|
|
63 |
"vidore/colqwen2-v0.1",
|
64 |
torch_dtype=torch.bfloat16,
|
65 |
device_map="cuda:0", # or "mps" if on Apple Silicon
|
66 |
-
)
|
67 |
processor = ColQwen2Processor.from_pretrained("vidore/colqwen2-v0.1")
|
68 |
|
69 |
# Your inputs
|
|
|
63 |
"vidore/colqwen2-v0.1",
|
64 |
torch_dtype=torch.bfloat16,
|
65 |
device_map="cuda:0", # or "mps" if on Apple Silicon
|
66 |
+
).eval()
|
67 |
processor = ColQwen2Processor.from_pretrained("vidore/colqwen2-v0.1")
|
68 |
|
69 |
# Your inputs
|