Bläddra i källkod

feat(adapter): 添加 GLM-OCR 适配器支持,增强模型适配器功能

zhch158_admin 1 vecka sedan
förälder
incheckning
371113b468
1 ändrade filer med 10 tillägg och 0 borttagningar
  1. 10 0
      ocr_tools/universal_doc_parser/models/adapters/__init__.py

+ 10 - 0
ocr_tools/universal_doc_parser/models/adapters/__init__.py

@@ -17,6 +17,13 @@ from .paddle_vl_adapter import PaddleVLRecognizer
 from .docling_layout_adapter import DoclingLayoutDetector
 from .pp_doclayout_v3_layout_adapter import PPDocLayoutV3Detector
 
+# GLM-OCR 适配器
+try:
+    from .glmocr_vl_adapter import GLMOCRVLRecognizer
+    GLMOCR_AVAILABLE = True
+except ImportError:
+    GLMOCR_AVAILABLE = False
+
 # 可选导入 DiT 适配器
 try:
     from .dit_layout_adapter import DitLayoutDetector
@@ -49,6 +56,9 @@ __all__ = [
     'PaddleLayoutDetector',
     'PaddleVLRecognizer',
     
+    # GLM-OCR 适配器
+    'GLMOCRVLRecognizer',
+    
     # Docling 适配器
     'DoclingLayoutDetector',
     # PP-DocLayoutV3 适配器