Update PyTorch version to fix vulnerability

biigle · Jul 26, 2024 · 90a555a · 90a555a
1 parent 0d0300a
commit 90a555a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,4 @@
-torch==2.1.*
+torch==2.2.*
 torchvision==0.16.*
 mmcv==2.1.*
 # Update CUDA Version if necessary.