Explorar el Código

hpi: add support for paddle311 (#4432)

zhang-prog hace 3 meses
padre
commit
550bbc058f
Se han modificado 2 ficheros con 4293 adiciones y 136 borrados
  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,

La diferencia del archivo ha sido suprimido porque es demasiado grande
+ 4287 - 135
paddlex/inference/utils/hpi_model_info_collection.json


Algunos archivos no se mostraron porque demasiados archivos cambiaron en este cambio