storytracer commited on
Commit
986835a
·
unverified ·
1 Parent(s): 0dc7702

Update olmOCR processor to Qwen2.5-VL-7B-Instruct

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ except Exception as e:
49
 
50
  # Load olmOCR
51
  try:
52
- PROCESSORS["olmOCR"] = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-7B-Instruct")
53
  MODELS["olmOCR"] = Qwen2_5_VLForConditionalGeneration.from_pretrained(
54
  "allenai/olmOCR-7B-0825", torch_dtype=torch.bfloat16, device_map="auto"
55
  )
 
49
 
50
  # Load olmOCR
51
  try:
52
+ PROCESSORS["olmOCR"] = AutoProcessor.from_pretrained("Qwen/Qwen2.5-VL-7B-Instruct")
53
  MODELS["olmOCR"] = Qwen2_5_VLForConditionalGeneration.from_pretrained(
54
  "allenai/olmOCR-7B-0825", torch_dtype=torch.bfloat16, device_map="auto"
55
  )