diff --git a/requirements.txt b/requirements.txt
index 1ee666c..e398772 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,2 +1,3 @@
 nats_bench>=1.4
-pytorch
+torchvision
+qlib  # For some modules