diff --git a/pyproject.toml b/pyproject.toml index 34cf7d2b..14b924d9 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -25,7 +25,7 @@ pytest = "7.4.2" einops = "0.7.0" bitsandbytes = "0.38.1" typing = "3.7.4.3" -transformers = "4.35.0" +transformers = "4.36.0" einops-exts = "0.0.4" torchvision = "*" accelerate = "0.22.0"