Quellcode durchsuchen

fix(paddle_vl_adapter): update model name retrieval logic to ensure default value is used

- Changed model name assignment to use config.get() for better flexibility.
- Ensured that the default model name 'PaddleOCR-VL-0.9B' is applied if not explicitly set in the configuration.
zhch158_admin vor 2 Tagen
Ursprung
Commit
d560cc01e0

+ 4 - 2
ocr_tools/universal_doc_parser/models/adapters/paddle_vl_adapter.py

@@ -29,7 +29,8 @@ class PaddleVLRecognizer(MinerUVLRecognizer):
     
     def __init__(self, config: Dict[str, Any]):
         # 🔧 强制设置 PaddleOCR-VL 模型名称
-        config['model_name'] = 'PaddleOCR-VL-0.9B'
+        # config['model_name'] = 'PaddleOCR-VL-0.9B'
+        config['model_name'] = config.get('model_name', 'PaddleOCR-VL-0.9B')
         
         # 🔧 确保使用正确的后端配置
         if config.get('backend') not in ['http-client']:
@@ -50,6 +51,7 @@ class PaddleVLRecognizer(MinerUVLRecognizer):
             backend = self.config.get('backend', 'http-client')
             server_url = self.config.get('server_url')
             model_params = self.config.get('model_params', {})
+            model_name = self.config.get('model_name', 'PaddleOCR-VL-0.9B')
             
             # 🔧 提取 MinerUClient 所需的参数
             # 从 model_params 中获取,如果没有则使用默认值
@@ -74,7 +76,7 @@ class PaddleVLRecognizer(MinerUVLRecognizer):
                 # HTTP客户端模式
                 self.vlm_model = MinerUClient(
                     backend=backend,
-                    model_name=self.config['model_name'],
+                    model_name=model_name,
                     server_url=server_url,
                     prompts=prompts,
                     max_concurrency=max_concurrency,