diff --git a/pyproject.toml b/pyproject.toml index e80053e..50f600c 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.1", "torchvision==0.16", "torchaudio==2.1.0" ] diff --git a/requirements.txt b/requirements.txt index 6b23c17..af5415c 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.1 torchvision==0.16 torchaudio==2.1.0