fix export with pir for OCR

paddlex/repo_apis/PaddleOCR_api/formula_rec/model.py

@@ -135,11 +135,9 @@ class FormulaRecModel(BaseModel):
         # PDX related settings
         device_type = device.split(":")[0]
         uniform_output_enabled = kwargs.pop("uniform_output_enabled", True)
-        export_with_pir = kwargs.pop("export_with_pir", False)
         config.update({"Global.uniform_output_enabled": uniform_output_enabled})
         config.update({"Global.model_name": self.name})
-        if export_with_pir:
-            config.update({"Global.export_with_pir": export_with_pir})
+        config.update({"Global.export_with_pir": kwargs.pop("export_with_pir", False)})
 
         self._assert_empty_kwargs(kwargs)
 
@@ -272,11 +270,9 @@ class FormulaRecModel(BaseModel):
 
         # PDX related settings
         uniform_output_enabled = kwargs.pop("uniform_output_enabled", True)
-        export_with_pir = kwargs.pop("export_with_pir", False)
         config.update({"Global.uniform_output_enabled": uniform_output_enabled})
         config.update({"Global.model_name": self.name})
-        if export_with_pir:
-            config.update({"Global.export_with_pir": export_with_pir})
+        config.update({"Global.export_with_pir": kwargs.pop("export_with_pir", False)})
 
         self._assert_empty_kwargs(kwargs)
 

paddlex/repo_apis/PaddleOCR_api/text_rec/model.py

@@ -135,11 +135,9 @@ class TextRecModel(BaseModel):
         # PDX related settings
         device_type = device.split(":")[0]
         uniform_output_enabled = kwargs.pop("uniform_output_enabled", True)
-        export_with_pir = kwargs.pop("export_with_pir", False)
         config.update({"Global.uniform_output_enabled": uniform_output_enabled})
         config.update({"Global.model_name": self.name})
-        if export_with_pir:
-            config.update({"Global.export_with_pir": export_with_pir})
+        config.update({"Global.export_with_pir": kwargs.pop("export_with_pir", False)})
 
         self._assert_empty_kwargs(kwargs)
 
@@ -272,11 +270,9 @@ class TextRecModel(BaseModel):
 
         # PDX related settings
         uniform_output_enabled = kwargs.pop("uniform_output_enabled", True)
-        export_with_pir = kwargs.pop("export_with_pir", False)
         config.update({"Global.uniform_output_enabled": uniform_output_enabled})
         config.update({"Global.model_name": self.name})
-        if export_with_pir:
-            config.update({"Global.export_with_pir": export_with_pir})
+        config.update({"Global.export_with_pir": kwargs.pop("export_with_pir", False)})
 
         self._assert_empty_kwargs(kwargs)