diff --git a/bird_cnn/requirements.txt b/bird_cnn/requirements.txt
index 5ba1753..1341ecd 100644
--- a/bird_cnn/requirements.txt
+++ b/bird_cnn/requirements.txt
@@ -1,7 +1,7 @@
 fastapi==0.136.1
 networkx==3.6.1
 numpy==2.3.4
-torch==2.11.0+cu126
-torchvision==0.26.0+cu126
+torch==2.1.0
+torchvision==0.26.0
 tqdm==4.67.3
 uvicorn==0.46.0