소스 검색

bugfix: some model don't support json format

gaotingquan 7 달 전
부모
커밋
bbed98d8cb
1개의 변경된 파일2개의 추가작업 그리고 0개의 파일을 삭제
  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"