瀏覽代碼

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: