diff --git a/setup.py b/setup.py index b9c62d789db703180f5fa6b25adc68a79366dce3..7170fb1c41fa6c77be3abf89cadb204e049e49b1 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',