diff --git a/setup.py b/setup.py index 55dc4fa353..f438dd8225 100644 --- a/setup.py +++ b/setup.py @@ -132,7 +132,7 @@ _deps = [ "librosa", "nltk", "natten>=0.14.6,<0.15.0", - "numpy>=1.17", + "numpy>=1.17,<2.0", "onnxconverter-common", "onnxruntime-tools>=1.4.2", "onnxruntime>=1.4.0", diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py index d699a6ac7f..7f8c3e4433 100644 --- a/src/transformers/dependency_versions_table.py +++ b/src/transformers/dependency_versions_table.py @@ -38,7 +38,7 @@ deps = { "librosa": "librosa", "nltk": "nltk", "natten": "natten>=0.14.6,<0.15.0", - "numpy": "numpy>=1.17", + "numpy": "numpy>=1.17,<2.0", "onnxconverter-common": "onnxconverter-common", "onnxruntime-tools": "onnxruntime-tools>=1.4.2", "onnxruntime": "onnxruntime>=1.4.0",