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

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"