diff --git a/pyproject.toml b/pyproject.toml index 2a1cf597..f4617c04 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -58,7 +58,7 @@ pytest = "8.3.2" torchfix = "*" einops = "0.8.0" bitsandbytes = "*" -transformers = "4.44.0" +transformers = "4.44.2" einops-exts = "0.0.4" torchvision = "*" accelerate = "0.33.0"