소스 검색

feat: 更新VL识别器创建逻辑,支持Paddle适配器

zhch158_admin 2 주 전
부모
커밋
b13ce17c58
1개의 변경된 파일4개의 추가작업 그리고 1개의 파일을 삭제
  1. 4 1
      zhch/universal_doc_parser/core/model_factory.py

+ 4 - 1
zhch/universal_doc_parser/core/model_factory.py

@@ -50,7 +50,10 @@ class ModelFactory:
         """创建VL识别器"""
         module_name = config.get('module', 'mineru')
         
-        if module_name == 'mineru':
+        if module_name == 'paddle':
+            from models.adapters import PaddleVLRecognizer
+            recognizer = PaddleVLRecognizer(config)
+        elif module_name == 'mineru':
             from models.adapters import MinerUVLRecognizer
             recognizer = MinerUVLRecognizer(config)
         else: