From 38f6f578e6ba2d3ea95780036ab5e215af307ef6 Mon Sep 17 00:00:00 2001
From: Guanghua Yu <742925032@qq.com>
Date: Wed, 12 Oct 2022 13:10:08 +0800
Subject: [PATCH] fix paddle2onnx version when export new format quant model
 (#1458)

---
 paddleslim/common/load_model.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/paddleslim/common/load_model.py b/paddleslim/common/load_model.py
index 2f31f97d..0b4bef2c 100644
--- a/paddleslim/common/load_model.py
+++ b/paddleslim/common/load_model.py
@@ -217,11 +217,11 @@ def export_onnx(model_dir,
     try:
         import paddle2onnx
         version = paddle2onnx.__version__
-        if version != '1.0.0rc4':
-            os.system('python -m pip install -U paddle2onnx==1.0.0rc4')
+        if version != '1.0.1':
+            os.system('python -m pip install -U paddle2onnx==1.0.1')
     except:
         from pip._internal import main
-        main(['install', 'paddle2onnx==1.0.0rc4'])
+        main(['install', 'paddle2onnx==1.0.1'])
     import paddle2onnx
     paddle2onnx.command.c_paddle_to_onnx(
         model_file=os.path.join(model_dir, model_filename),
-- 
GitLab