Pārlūkot izejas kodu

hpi: add support for paddle311 (#4432)

zhang-prog 3 mēneši atpakaļ
vecāks
revīzija
550bbc058f

+ 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,

Failā izmaiņas netiks attēlotas, jo tās ir par lielu
+ 4287 - 135
paddlex/inference/utils/hpi_model_info_collection.json


Daži faili netika attēloti, jo izmaiņu fails ir pārāk liels