From ea82404a6dcb9d04396bb3928a4a08b2da78491f Mon Sep 17 00:00:00 2001 From: Yuting Jiang Date: Tue, 13 Dec 2022 14:00:04 +0000 Subject: [PATCH] downgrage transformers version to fix tersorrt --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index b263997c6..1e8f67492 100644 --- a/setup.py +++ b/setup.py @@ -204,7 +204,7 @@ def run(self): 'torch': [ 'torch>=1.7.0a0', 'torchvision>=0.8.0a0', - 'transformers>=4.3.3', + 'transformers>=4.3.3, <4.23.0', ], 'ort': [ 'onnx>=1.10.2',