From 2b538794926711f89f746390098fea7808d29e9b Mon Sep 17 00:00:00 2001
From: truthless11 <truthless11@gmail.com>
Date: Tue, 8 Sep 2020 19:26:56 +0800
Subject: [PATCH] update versions
---
setup.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/setup.py b/setup.py
index b9c62d7..7170fb1 100755
--- a/setup.py
+++ b/setup.py
@@ -38,12 +38,12 @@ setup(
'visdom',
'Pillow',
'future',
- 'torch',
+ 'torch>=1.2.0,<1.6.0',
'numpy>=1.15.0',
'scipy',
'scikit-learn==0.20.3',
'pytorch-pretrained-bert>=0.6.1',
- 'transformers>=2.3.0',
+ 'transformers>=2.3.0,<3.0.0',
'tensorflow==1.14',
'tensorboard>=1.14.0',
'tensorboardX==1.7',
--
GitLab