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