diff --git a/pyproject.toml b/pyproject.toml index e80053e..3b40021 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -5,7 +5,7 @@ description = 'The Microsoft MX floating point library' readme = "README.md" requires-python = ">=3.8" dependencies = [ - "torch==2.2.0", + "torch==2.7.0", "torchvision==0.16", "torchaudio==2.1.0" ] diff --git a/requirements.txt b/requirements.txt index 6b23c17..f4240dd 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ # Most recent tested torch version, older versions # will probably work. -torch==2.2.0 +torch==2.7.0 torchvision==0.16 torchaudio==2.1.0