diff --git a/pyproject.toml b/pyproject.toml index c293c290..6044b51c 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -56,7 +56,7 @@ python = "^3.10" torch = ">=2.1.1,<3.0" einops = "0.8.0" bitsandbytes = "*" -transformers = "4.44.2" +transformers = "4.47.0" einops-exts = "0.0.4" torchvision = "*" accelerate = "0.33.0"