Update requirements.txt

requirements.txt

@@ -1,5 +1,3 @@
-
-
 absl-py==2.1.0
 aiofiles==23.2.1
 aiohappyeyeballs==2.4.3
@@ -142,7 +140,7 @@ pycparser==2.22
 pydantic==2.9.2
 pydantic_core==2.23.4
 pydub==0.25.1
-
+pyg-lib==0.4.0+pt24cu121
 Pygments==2.18.0
 pyparsing==3.2.0
 python-dateutil==2.9.0.post0
@@ -188,12 +186,12 @@ threadpoolctl==3.5.0
 tinycss2==1.4.0
 tomli==2.0.2
 tomlkit==0.12.0
-torch
+torch==2.5.1
 torch-geometric==2.6.1
-torch_cluster==1.6.3
-torch_scatter==2.1.2
-torch_sparse==0.6.18
-torch_spline_conv==1.2.2
+torch_cluster==1.6.3+pt24cu124
+torch_scatter==2.1.2+pt24cu124
+torch_sparse==0.6.18+pt24cu124
+torch_spline_conv==1.2.2+pt24cu124
 torchmetrics==1.5.1
 torchtyping==0.1.5
 tornado==6.4.1