Safetensors
qwen2_vl
vidore
reranker
NohTow commited on
Commit
9205513
1 Parent(s): 6d48017

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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-2B-LoRA-Reranker",
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 = "Is this your query about a document ?"
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