diff --git a/requirements.txt b/requirements.txt index 99a76a076..b1bcf36ee 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,3 @@ -r requirements-no-torch.txt -torch>=1.5.0 -torchvision==0.6.0 +torch>=1.8.1 +torchvision==0.9.1 diff --git a/setup.py b/setup.py index c151cea58..1e426a3ee 100644 --- a/setup.py +++ b/setup.py @@ -73,10 +73,10 @@ "scipy == 1.4.1", "sentencepiece == 0.1.86", "tokenizers == 0.8.1.rc2", - "torch >= 1.5.0", + "torch >= 1.8.1", "tqdm == 4.46.0", "transformers == 3.1.0", - "torchvision == 0.6.0", + "torchvision == 0.9.1", ], extras_require=extras, python_requires=">=3.6.0",