diff --git a/setup.py b/setup.py
index 1672c3a..1244b5d 100644
--- a/setup.py
+++ b/setup.py
@@ -25,6 +25,7 @@ setup(
     'click',
     'einops>=0.4',
     'einops-exts',
+    'pillow',
     'torch>=1.10',
     'torchvision',
     'x-clip>=0.4.1',