diff --git a/pyproject.toml b/pyproject.toml index 4740a4c1..4bb37387 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -19,7 +19,7 @@ packages = [ python = "^3.8" torch = "*" fairscale = "*" -timm = "0.4.12" +timm = "0.9.7" pytest = "*" einops = "*" bitsandbytes = "*"