diff --git a/pyproject.toml b/pyproject.toml index d7a58b36..33ddc40d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -58,7 +58,7 @@ pytest = "8.2.2" torchfix = "*" einops = "0.8.0" bitsandbytes = "*" -transformers = "4.42.3" +transformers = "4.44.0" einops-exts = "0.0.4" torchvision = "*" accelerate = "0.33.0"