Added lion-pytorch dependency to setup.py

ZeroCool940711 · Mar 9, 2023 · 55a7d80 · 55a7d80
1 parent 8677417
commit 55a7d80
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -34,6 +34,7 @@
         "torchvision",
         "tqdm",
         "vector-quantize-pytorch>=0.10.14",
+        "lion-pytorch",
     ],
     classifiers=[
         "Development Status :: 4 - Beta",