diff --git a/pyproject.toml b/pyproject.toml index 7ef7ee3b..94aff05c 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -53,7 +53,7 @@ dependencies = [ "dgl<=2.2.1", "pymatgen", "lightning<=2.5.0.post0", - "torch<=2.2.1", + "torch<=2.6.0", "pydantic", "torchdata<=0.7.1", "boto3", diff --git a/requirements.txt b/requirements.txt index 082f1d32..1c456519 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ dgl==2.2.1 -torch==2.2.1 +torch==2.6.0 lightning==2.5.0.post0 pymatgen==2024.11.13 ase==3.24.0