ocr-time-machine

Running on Zero

storytracer commited on Aug 29

Commit

0dc7702

unverified ·

1 Parent(s): 15f06e4

Run olmOCR with Qwen2_5_VLForConditionalGeneration

Files changed (1) hide show

app.py CHANGED Viewed

@@ -50,7 +50,7 @@ except Exception as e:
 # Load olmOCR
 try:
     PROCESSORS["olmOCR"] = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-7B-Instruct")
-    MODELS["olmOCR"] = Qwen2VLForConditionalGeneration.from_pretrained(
         "allenai/olmOCR-7B-0825", torch_dtype=torch.bfloat16, device_map="auto"
     )
     PIPELINES["olmOCR"] = pipeline(

 # Load olmOCR
 try:
     PROCESSORS["olmOCR"] = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-7B-Instruct")
+    MODELS["olmOCR"] = Qwen2_5_VLForConditionalGeneration.from_pretrained(
         "allenai/olmOCR-7B-0825", torch_dtype=torch.bfloat16, device_map="auto"
     )
     PIPELINES["olmOCR"] = pipeline(