diff --git a/pyproject.toml b/pyproject.toml index e5c9d7b..916b984 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -12,6 +12,7 @@ requires = [ "scipy", "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 0947fe3..999d2ce 100644 --- a/requirements.txt +++ b/requirements.txt @@ -26,3 +26,5 @@ numba #--pre #--extra-index-url https://mirrors.huaweicloud.com/ascend/repos/pypi torch-npu==2.7.1 + +transformers<=4.57.1