Spaces:

prithivMLmods
/

Multimodal-OCR3

Running on Zero

prithivMLmods commited on Nov 6

Commit

0e197cb

verified ·

1 Parent(s): 515bb0e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -174,7 +174,7 @@ processor_x = AutoProcessor.from_pretrained(MODEL_ID_X, trust_remote_code=True)
 model_x = Qwen2_5_VLForConditionalGeneration.from_pretrained(
     MODEL_ID_X,
     trust_remote_code=True,
-    torch_dtype=torch.float16
 ).to(device).eval()
 # Load Dots.OCR from the local, patched directory

 model_x = Qwen2_5_VLForConditionalGeneration.from_pretrained(
     MODEL_ID_X,
     trust_remote_code=True,
+    torch_dtype="auto",
 ).to(device).eval()
 # Load Dots.OCR from the local, patched directory