|
|
@@ -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,
|