diff --git a/requirements.txt b/requirements.txt
index da1fbb27..e804618e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,6 @@
 torch
 torchsde
+torchvision
 einops
 transformers>=4.25.1
 safetensors>=0.3.0