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

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


Някои файлове не бяха показани, защото твърде много файлове са промени