From 76c767b1ce082255ef7ca9850a03190cd8b9c467 Mon Sep 17 00:00:00 2001
From: Phil Wang <lucidrains@gmail.com>
Date: Sun, 1 May 2022 12:22:15 -0700
Subject: [PATCH] update deps, commit to using webdatasets, per @rom1504
 consultation

---
 setup.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index 9b27a98..eb52bf3 100644
--- a/setup.py
+++ b/setup.py
@@ -26,15 +26,16 @@ setup(
     'clip-anytorch',
     'einops>=0.4',
     'einops-exts>=0.0.3',
+    'embedding-reader',
     'kornia>=0.5.4',
     'pillow',
     'torch>=1.10',
     'torchvision',
     'tqdm',
     'vector-quantize-pytorch',
+    'webdataset',
     'x-clip>=0.5.1',
-    'youtokentome',
-    'embedding-reader'
+    'youtokentome'
   ],
   classifiers=[
     'Development Status :: 4 - Beta',