Skip to content

Commit

Permalink
add onnx dependency for x2paddle (#1160)
Browse files Browse the repository at this point in the history
* add denpendency

* add onnx dependency

* add onnx requirement
  • Loading branch information
rainyfly authored Jan 9, 2023
1 parent 7bde440 commit 9a16e46
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 1 deletion.
1 change: 1 addition & 0 deletions requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -12,3 +12,4 @@ multiprocess
packaging
x2paddle
rarfile
onnx >= 1.6.0
2 changes: 1 addition & 1 deletion visualdl/component/inference/model_convert_server.py
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,7 @@ def convert_model(self, format):
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':
Expand Down

0 comments on commit 9a16e46

Please sign in to comment.