diff --git a/setup.py b/setup.py
index b3f02bb..d6a2424 100644
--- a/setup.py
+++ b/setup.py
@@ -26,6 +26,7 @@ setup(
   install_requires=[
     'accelerate',
     'click',
+    'open-clip-torch>=2.0.0,<3.0.0',
     'clip-anytorch>=2.4.0',
     'coca-pytorch>=0.0.5',
     'ema-pytorch>=0.0.7',