diff --git a/setup.py b/setup.py
index 59f0ed3aea..83cf6eed49 100644
--- a/setup.py
+++ b/setup.py
@@ -79,7 +79,8 @@
         'redis',  # parallel
         'mpire>=2.3.5',  # parallel
         'einops',
-        'loguru'
+        'loguru',
+        'vllm'
     ],
     extras_require={
         'test': [