소스 검색

Fix serving schema and doc vlm predictor (#4008)

Lin Manhui 6 달 전
부모
커밋
53a0aa4892
2개의 변경된 파일10개의 추가작업 그리고 9개의 파일을 삭제
  1. 7 6
      paddlex/inference/models/doc_vlm/predictor.py
  2. 3 3
      paddlex/inference/serving/schemas/pp_structurev3.py

+ 7 - 6
paddlex/inference/models/doc_vlm/predictor.py

@@ -14,6 +14,7 @@
 
 import copy
 import os
+import warnings
 from typing import List
 
 from ....modules.doc_vlm.model_list import MODELS
@@ -87,8 +88,8 @@ class DocVLMPredictor(BasePredictor):
         # build model
         if self.model_name in self.model_group["PP-DocBee"]:
             if kwargs.get("use_hpip", False):
-                raise ValueError(
-                    f"PP-DocBee series do not support `use_hpip=True` for now."
+                warnings.warn(
+                    "The PP-DocBee series does not support `use_hpip=True` for now."
                 )
             with TemporaryDeviceChanger(self.device):
                 model = PPDocBeeInference.from_pretrained(
@@ -96,8 +97,8 @@ class DocVLMPredictor(BasePredictor):
                 )
         elif self.model_name in self.model_group["PP-Chart2Table"]:
             if kwargs.get("use_hpip", False):
-                raise ValueError(
-                    f"PP-Chart2Table series do not support `use_hpip=True` for now."
+                warnings.warn(
+                    "The PP-Chart2Table series does not support `use_hpip=True` for now."
                 )
             with TemporaryDeviceChanger(self.device):
                 model = PPChart2TableInference.from_pretrained(
@@ -107,8 +108,8 @@ class DocVLMPredictor(BasePredictor):
                 )
         elif self.model_name in self.model_group["PP-DocBee2"]:
             if kwargs.get("use_hpip", False):
-                raise ValueError(
-                    f"PP-Chart2Table series do not support `use_hpip=True` for now."
+                warnings.warn(
+                    "The PP-Chart2Table series does not support `use_hpip=True` for now."
                 )
             with TemporaryDeviceChanger(self.device):
                 model = PPDocBee2Inference.from_pretrained(

+ 3 - 3
paddlex/inference/serving/schemas/pp_structurev3.py

@@ -32,13 +32,13 @@ INFER_ENDPOINT: Final[str] = "/layout-parsing"
 
 
 class InferRequest(ocr.BaseInferRequest):
-    useDocOrientationClassify: Optional[bool] = None
-    useDocUnwarping: Optional[bool] = None
+    useDocOrientationClassify: Optional[bool] = False
+    useDocUnwarping: Optional[bool] = False
     useTextlineOrientation: Optional[bool] = None
     useSealRecognition: Optional[bool] = None
     useTableRecognition: Optional[bool] = None
     useFormulaRecognition: Optional[bool] = None
-    useChartRecognition: Optional[bool] = None
+    useChartRecognition: Optional[bool] = False
     useRegionDetection: Optional[bool] = None
     layoutThreshold: Optional[float] = None
     layoutNms: Optional[bool] = None