diff --git a/setup.py b/setup.py index 922258d65efab7..2dd0c50fb8d693 100644 --- a/setup.py +++ b/setup.py @@ -179,7 +179,7 @@ "tf2onnx", "timeout-decorator", "tiktoken", - "timm<=0.9.16", + "timm<=1.0.11", "tokenizers>=0.20,<0.21", "torch", "torchaudio", diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py index 9543b58ad40d91..dcf216783befaa 100644 --- a/src/transformers/dependency_versions_table.py +++ b/src/transformers/dependency_versions_table.py @@ -84,7 +84,7 @@ "tf2onnx": "tf2onnx", "timeout-decorator": "timeout-decorator", "tiktoken": "tiktoken", - "timm": "timm<=0.9.16", + "timm": "timm<=1.0.11", "tokenizers": "tokenizers>=0.20,<0.21", "torch": "torch", "torchaudio": "torchaudio",