diff --git a/requirements.txt b/requirements.txt
index 1a14425..f5cd155 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -27,5 +27,5 @@ wandb==0.18.3
 streamlit==1.50.0
 einops==0.8.1
 swanlab==0.6.8
-# torch==2.3.0
-# torchvision==0.18.0
\ No newline at end of file
+torch>=2.6.0
+torchvision>=0.21.0
\ No newline at end of file