ソースを参照

hpi: add support for paddle311 (#4432)

zhang-prog 3 ヶ月 前
コミット
550bbc058f
2 ファイル変更4293 行追加136 行削除
  1. 6 1
      paddlex/inference/utils/hpi.py
  2. 4287 135
      paddlex/inference/utils/hpi_model_info_collection.json

+ 6 - 1
paddlex/inference/utils/hpi.py

@@ -157,7 +157,12 @@ def suggest_inference_backend_and_config(
 
     paddle_version = get_paddle_version()
     if (3, 0) <= paddle_version[:2] <= (3, 1) and paddle_version[3] is None:
-        paddle_version = f"paddle{paddle_version[0]}{paddle_version[1]}"
+        if paddle_version[2] == 0:
+            paddle_version = f"paddle{paddle_version[0]}{paddle_version[1]}"
+        else:
+            paddle_version = (
+                f"paddle{paddle_version[0]}{paddle_version[1]}{paddle_version[2]}"
+            )
     else:
         return (
             None,

ファイルの差分が大きいため隠しています
+ 4287 - 135
paddlex/inference/utils/hpi_model_info_collection.json


この差分においてかなりの量のファイルが変更されているため、一部のファイルを表示していません