Răsfoiți Sursa

bugfix: some model don't support json format

gaotingquan 7 luni în urmă
părinte
comite
bbed98d8cb
1 a modificat fișierele cu 2 adăugiri și 0 ștergeri
  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"