Parcourir la source

hpi: add support for paddle311 (#4432)

zhang-prog il y a 3 mois
Parent
commit
550bbc058f
2 fichiers modifiés avec 4293 ajouts et 136 suppressions
  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,

Fichier diff supprimé car celui-ci est trop grand
+ 4287 - 135
paddlex/inference/utils/hpi_model_info_collection.json


Certains fichiers n'ont pas été affichés car il y a eu trop de fichiers modifiés dans ce diff