From 9a16e460acd0158a186b923dcbc1ce5a4028d721 Mon Sep 17 00:00:00 2001
From: chenjian <chenjian26@baidu.com>
Date: Mon, 9 Jan 2023 14:26:41 +0800
Subject: [PATCH] add onnx dependency for x2paddle  (#1160)

* add denpendency

* add onnx dependency

* add onnx requirement
---
 requirements.txt                                     | 1 +
 visualdl/component/inference/model_convert_server.py | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index 12588e37..faef06cf 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -12,3 +12,4 @@ multiprocess
 packaging
 x2paddle
 rarfile
+onnx >= 1.6.0
\ No newline at end of file
diff --git a/visualdl/component/inference/model_convert_server.py b/visualdl/component/inference/model_convert_server.py
index 6f66dd5c..e9dfb36f 100644
--- a/visualdl/component/inference/model_convert_server.py
+++ b/visualdl/component/inference/model_convert_server.py
@@ -59,7 +59,7 @@ class ModelConvertApi(object):
                             import onnx  # noqa: F401
                         except Exception:
                             raise RuntimeError(
-                                "[ERROR] onnx is not installed, use \"pip install onnx==1.6.0\"."
+                                "[ERROR] onnx is not installed, use \"pip install onnx>=1.6.0\"."
                             )
                         onnx2paddle(fp.name, tmpdirname)
                     elif format == 'caffe':
-- 
GitLab