diff --git a/pyproject.toml b/pyproject.toml index 5b659cd30..ab034ff2f 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -14,7 +14,7 @@ requires-python = ">=3.8" license = { file = "LICENSE" } dependencies = [ "numpy", - "torch>=2.1,<2.3", + "torch>=2.1,<2.4", "ai2-olmo-core==0.1.0", "omegaconf", "rich",