Spaces:
Running
on
Zero
Running
on
Zero
Update olmOCR processor to Qwen2.5-VL-7B-Instruct
Browse files
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 |
)
|