diff --git a/pyproject.toml b/pyproject.toml index 4696d18..6190d2b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -24,7 +24,7 @@ dynamic = ["version"] dependencies = [ "numpy>=1.26.4,<2.3.0", "accelerate>=0.20.3,!=0.34,<1.11", -"transformers>4.45,<5.9", +"transformers>4.45,<=5.12.1", "torch>=2.2.0,<2.12.0", "tqdm>=4.66.2,<5.0", "datasets>=3.0.0,<5.0",