diff --git a/setup.py b/setup.py index f7e31559acbe..b9915f0076ec 100644 --- a/setup.py +++ b/setup.py @@ -144,7 +144,7 @@ "protobuf", "psutil", "pyyaml>=5.1", - "pydantic<2", + "pydantic", "pytest>=7.2.0", "pytest-timeout", "pytest-xdist", diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py index cecdaf3419fc..5f5883c2e4bd 100644 --- a/src/transformers/dependency_versions_table.py +++ b/src/transformers/dependency_versions_table.py @@ -50,7 +50,7 @@ "protobuf": "protobuf", "psutil": "psutil", "pyyaml": "pyyaml>=5.1", - "pydantic": "pydantic<2", + "pydantic": "pydantic", "pytest": "pytest>=7.2.0", "pytest-timeout": "pytest-timeout", "pytest-xdist": "pytest-xdist",