diff --git a/requirements.txt b/requirements.txt index cd9e66c..b21f2c2 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,7 +5,7 @@ lightning torchmetrics>=0.11.4 datasets transformers>=4.41.0 -torch_geometric==2.3.0 +torch_geometric==2.7.0 --extra-index-url https://download.pytorch.org/whl/cu121 torch_scatter torch_sparse