From 9650ce42b046a1ef61338479a0fefadf3365deb2 Mon Sep 17 00:00:00 2001
From: Guanghua Yu <742925032@qq.com>
Date: Wed, 28 Sep 2022 11:57:42 +0800
Subject: [PATCH] fix x2paddle version import (#1437)

---
 paddleslim/common/load_model.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/paddleslim/common/load_model.py b/paddleslim/common/load_model.py
index e611628b..f4f3be28 100644
--- a/paddleslim/common/load_model.py
+++ b/paddleslim/common/load_model.py
@@ -121,9 +121,16 @@ def load_onnx_model(model_path,
             model_path), 'Not found `{}`, please check model path.'.format(
                 model_path)
         try:
-            pkg.require('x2paddle')
+            import x2paddle
+            version = x2paddle.__version__
+            v0, v1, v2 = version.split('.')
+            version_sum = int(v0) * 100 + int(v1) * 10 + int(v2)
+            if version_sum < 139:
+                _logger.error(
+                    "x2paddle>=1.3.9 is required, please use \"pip install x2paddle\"."
+                )
         except:
-            os.system('python -m pip install -U x2paddle==1.3.9')
+            os.system('python -m pip install -U x2paddle')
         # check onnx installation and version
         try:
             pkg.require('onnx')
-- 
GitLab