From 5a3d0cbdda73afafebbb8516901e622242a67863 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger Date: Tue, 28 Jun 2022 15:07:26 -0400 Subject: [PATCH] Pin PyTorch while we fix compatibility with 1.12 --- setup.py | 2 +- src/transformers/dependency_versions_table.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 36865a5051..ea8037d4bb 100644 --- a/setup.py +++ b/setup.py @@ -160,7 +160,7 @@ _deps = [ "timeout-decorator", "timm", "tokenizers>=0.11.1,!=0.11.3,<0.13", - "torch>=1.0", + "torch>=1.0,<1.12", "torchaudio", "pyctcdecode>=0.3.0", "tqdm>=4.27", diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py index 41cf2cba5b..d63b79abab 100644 --- a/src/transformers/dependency_versions_table.py +++ b/src/transformers/dependency_versions_table.py @@ -66,7 +66,7 @@ deps = { "timeout-decorator": "timeout-decorator", "timm": "timm", "tokenizers": "tokenizers>=0.11.1,!=0.11.3,<0.13", - "torch": "torch>=1.0", + "torch": "torch>=1.0,<1.12", "torchaudio": "torchaudio", "pyctcdecode": "pyctcdecode>=0.3.0", "tqdm": "tqdm>=4.27",