diff --git a/diffraction/WISH/bragg-detect/cnn/requirements.txt b/diffraction/WISH/bragg-detect/cnn/requirements.txt
index 7a6670b..e382e8a 100644
--- a/diffraction/WISH/bragg-detect/cnn/requirements.txt
+++ b/diffraction/WISH/bragg-detect/cnn/requirements.txt
@@ -1,6 +1,4 @@
--f https://download.pytorch.org/whl/cu118
-torch
-torchvision
-
+torch==2.5.1
+torchvision==0.20.1
 albumentations==1.4.0
 tqdm==4.66.3