From 7ad4631fc259aadd2fae827882ddc25cb7f11b5b Mon Sep 17 00:00:00 2001 From: Always-Naive <97138029+Always-Naive@users.noreply.github.com> Date: Mon, 19 May 2025 14:31:17 +0800 Subject: [PATCH] fix version with suffix --- deploy/utils/predictor.py | 1 + 1 file changed, 1 insertion(+) diff --git a/deploy/utils/predictor.py b/deploy/utils/predictor.py index 04f9eab7a8..4fa5b3d49e 100644 --- a/deploy/utils/predictor.py +++ b/deploy/utils/predictor.py @@ -57,6 +57,7 @@ def create_paddle_predictor(self, args, inference_model_dir=None): # NOTE: paddle support to PIR mode after v2.6.0 pd_version = 0 for v in paddle.__version__.split(".")[:3]: + v = v.split('-')[0] pd_version = 10 * pd_version + eval(v) if pd_version == 0 or pd_version >= 260: