diff --git a/setup.py b/setup.py
index 39686e2..aa90fb3 100644
--- a/setup.py
+++ b/setup.py
@@ -22,6 +22,8 @@
             "open_clip_torch",
             "wget",
             "scipy",
+            "datasets",
+            "transformers",
         ],
         "fast": ["fast_jl"],
     },