diff --git a/pyproject.toml b/pyproject.toml index 1e184f39..fca70490 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -53,7 +53,7 @@ dependencies = [ "dgl<=2.4.0", "pymatgen", "lightning<=2.4.0", - "torch<=2.2.1", + "torch<=2.4.1", "pydantic", "torchdata<=0.7.1", "boto3", diff --git a/requirements.txt b/requirements.txt index e8bec381..4a679bf6 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ dgl==2.4.0 -torch==2.4.0 +torch==2.4.1 lightning==2.4.0 pymatgen==2024.9.17.1 ase==3.23.0