feat: update torch 2.5.1

scil-vital · Dec 18, 2024 · 634e58a · 634e58a
1 parent 20c76d7
commit 634e58a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -14,7 +14,7 @@ dipy==1.9.*
 # Other important dependencies
 # -------
 bertviz==1.4.0  # For transformer's visu
-torch==2.2.0
+torch==2.5.1
 tqdm==4.64.*
 comet-ml>=3.22.0
 contextlib2==21.6.0