diff --git a/setup.py b/setup.py
index d1979faf63..2b310bcfd0 100644
--- a/setup.py
+++ b/setup.py
@@ -115,7 +115,7 @@
 
 extra_deps['openai'] = [
     'openai==1.3.8',
-    'tiktoken==0.4.0',
+    'tiktoken>=0.4,<0.8.1',
 ]
 
 extra_deps['megablocks'] = [