fix: remove torch_dtype to fix TrOCR dtype mismatch
Browse files🤖 Generated with [Claude Code](https://claude.com/claude-code)
Co-Authored-By: Claude <noreply@anthropic.com>
app.py
CHANGED
|
@@ -20,7 +20,6 @@ def _load_ocr_model():
|
|
| 20 |
model = VisionEncoderDecoderModel.from_pretrained(
|
| 21 |
OCR_MODEL_ID,
|
| 22 |
device_map="auto",
|
| 23 |
-
torch_dtype=torch.float16,
|
| 24 |
)
|
| 25 |
|
| 26 |
processor = TrOCRProcessor.from_pretrained(OCR_MODEL_ID)
|
|
|
|
| 20 |
model = VisionEncoderDecoderModel.from_pretrained(
|
| 21 |
OCR_MODEL_ID,
|
| 22 |
device_map="auto",
|
|
|
|
| 23 |
)
|
| 24 |
|
| 25 |
processor = TrOCRProcessor.from_pretrained(OCR_MODEL_ID)
|