diff --git a/pyproject.toml b/pyproject.toml index 13947c1d..adfa0517 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -22,7 +22,7 @@ pytest = "8.1.1" torchfix = "*" einops = "0.7.0" bitsandbytes = "0.43.0" -transformers = "4.40.1" +transformers = "4.41.0" einops-exts = "0.0.4" torchvision = "0.18.0" accelerate = "0.28.0"