Преглед на файлове

fix pipline model_name input (#2864)

zhangyubo0722 преди 10 месеца
родител
ревизия
8f99cccf95
променени са 1 файла, в които са добавени 3 реда и са изтрити 5 реда
  1. 3 5
      paddlex/inference/pipelines_new/base.py

+ 3 - 5
paddlex/inference/pipelines_new/base.py

@@ -81,14 +81,12 @@ class BasePipeline(ABC, metaclass=AutoRegisterABCMetaClass):
         if "model_config_error" in config:
             raise ValueError(config["model_config_error"])
 
-        model_dir = config["model_dir"]
-        if model_dir == None:
-            model_dir = config["model_name"]
-
+        model_dir = config.get("model_dir", None)
         from .. import create_predictor
 
         model = create_predictor(
-            model=model_dir,
+            model_name=config["model_name"],
+            model_dir=model_dir,
             device=self.device,
             pp_option=self.pp_option,
             use_hpip=self.use_hpip,