Skip to content

GitLab

Unverified Commit 1deb2eaa authored by Yuting Jiang's avatar Yuting Jiang Committed by GitHub
Browse files

downgrage transformers version to fix tersorrt (#441) 

**Description**
Downgrage transformers version to fix tersorrt test failure.
parent e4eeda0a
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
setup.py
View file @ 1deb2eaa
...@@ -204,7 +204,7 @@ def run(self): ...@@ -204,7 +204,7 @@ def run(self):
'torch': [ 'torch': [
'torch>=1.7.0a0', 'torch>=1.7.0a0',
'torchvision>=0.8.0a0', 'torchvision>=0.8.0a0',
'transformers>=4.3.3', 'transformers>=4.3.3, <4.23.0',
], ],
'ort': [ 'ort': [
'onnx>=1.10.2', 'onnx>=1.10.2',
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment