diff --git a/requirements.txt b/requirements.txt
index 9ee90ca4..813a4772 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -9,6 +9,7 @@ numpy==1.26.2
 scipy==1.12.0
 requests==2.31.0
 torchsde==0.2.6
+torchvision==0.22.0
 
 einops
 opencv-contrib-python