diff --git a/pyproject.toml b/pyproject.toml index 7f90b1ed..1fa9d15f 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -14,6 +14,7 @@ requires = [ "pandas-stubs", "setuptools>=64", "setuptools-scm>=8", + "transformers<=4.57.1", "torch-npu==2.7.1", "torch==2.7.1", "torchvision", diff --git a/requirements.txt b/requirements.txt index 936de5f6..41a14390 100644 --- a/requirements.txt +++ b/requirements.txt @@ -28,3 +28,5 @@ numba #--pre #--extra-index-url https://mirrors.huaweicloud.com/ascend/repos/pypi torch-npu==2.7.1 + +transformers<=4.57.1