mirror/computer
1
0
Fork 0
mirror of https://github.com/trycua/computer.git synced 2026-01-10 07:20:10 -06:00
Code Issues Packages Projects Releases Wiki Activity

enabled MPS support for easyOCR

Browse Source
This commit is contained in:
Dillon DuPont
2025-04-04 09:22:16 -04:00
parent 0836bb7e8f
commit 6815d8a182
1 changed files with 2 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
libs/som/som/ocr.py
View File

@@ -67,17 +67,9 @@ class OCRProcessor:
import easyocr
# Use GPU if available
use_gpu = self.device in ["cuda"] # MPS not directly supported by EasyOCR
# If using MPS, add warnings to explain why CPU is used
if self.device == "mps":
logger.warning("EasyOCR doesn't support MPS directly. Using CPU instead.")
logger.warning(
"To silence this warning, set environment variable: PYTORCH_ENABLE_MPS_FALLBACK=1"
)
use_gpu = self.device in ["cuda", "mps"]
self.reader = easyocr.Reader(["en"], gpu=use_gpu)
# Verify reader initialization
if self.reader is None:
raise ValueError("Failed to initialize EasyOCR reader")