Update README.md
Browse files
README.md
CHANGED
@@ -25,14 +25,14 @@ from transformers import AutoProcessor, Qwen2VLForConditionalGeneration
|
|
25 |
# Load processor and model
|
26 |
processor = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-2B-Instruct")
|
27 |
model = Qwen2VLForConditionalGeneration.from_pretrained(
|
28 |
-
"lightonai/MonoQwen2-VL-
|
29 |
device_map="auto",
|
30 |
# attn_implementation="flash_attention_2",
|
31 |
# torch_dtype=torch.bfloat16,
|
32 |
)
|
33 |
|
34 |
# Define query and load image
|
35 |
-
query = "
|
36 |
image_path = "your/path/to/image.png"
|
37 |
image = Image.open(image_path)
|
38 |
|
|
|
25 |
# Load processor and model
|
26 |
processor = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-2B-Instruct")
|
27 |
model = Qwen2VLForConditionalGeneration.from_pretrained(
|
28 |
+
"lightonai/MonoQwen2-VL-v0.1",
|
29 |
device_map="auto",
|
30 |
# attn_implementation="flash_attention_2",
|
31 |
# torch_dtype=torch.bfloat16,
|
32 |
)
|
33 |
|
34 |
# Define query and load image
|
35 |
+
query = "What is ColPali?"
|
36 |
image_path = "your/path/to/image.png"
|
37 |
image = Image.open(image_path)
|
38 |
|