Quellcode durchsuchen

update hpi build_paddle_infer (#4354)

zhang-prog vor 4 Monaten
Ursprung
Commit
3a539df07f
1 geänderte Dateien mit 7 neuen und 1 gelöschten Zeilen
  1. 7 1
      paddlex/inference/models/common/static_infer.py

+ 7 - 1
paddlex/inference/models/common/static_infer.py

@@ -723,9 +723,15 @@ class HPInfer(StaticInfer):
                 )
                 kwargs["trt_dynamic_shape_input_data"] = trt_dynamic_shape_input_data
         pp_option = PaddlePredictorOption(**kwargs)
+        pp_option.setdefault_by_model_name(model_name=self._config.pdx_model_name)
         logging.info("Using Paddle Inference backend")
         logging.info("Paddle predictor option: %s", pp_option)
-        return PaddleInfer(self._model_dir, self._model_file_prefix, option=pp_option)
+        return PaddleInfer(
+            self._config.pdx_model_name,
+            self._model_dir,
+            self._model_file_prefix,
+            option=pp_option,
+        )
 
     def _build_ui_runtime(self, backend, backend_config, ui_option=None):
         # TODO: Validate the compatibility of backends with device types