Ver código fonte

bugfix: some model don't support json format

gaotingquan 7 meses atrás
pai
commit
bbed98d8cb
1 arquivos alterados com 2 adições e 0 exclusões
  1. 2 0
      paddlex/inference/utils/model_paths.py

+ 2 - 0
paddlex/inference/utils/model_paths.py

@@ -34,6 +34,8 @@ def get_model_paths(
     if FLAGS_json_format_model:
         if (model_dir / f"{model_file_prefix}.json").exists():
             pd_model_path = model_dir / f"{model_file_prefix}.json"
+        elif (model_dir / f"{model_file_prefix}.pdmodel").exists():
+            pd_model_path = model_dir / f"{model_file_prefix}.pdmodel"
     else:
         if (model_dir / f"{model_file_prefix}.json").exists():
             pd_model_path = model_dir / f"{model_file_prefix}.json"