From 55a7d80711a14d7330f6acffbd87d4b5b214a8cf Mon Sep 17 00:00:00 2001 From: ZeroCool940711 Date: Wed, 8 Mar 2023 17:51:30 -0700 Subject: [PATCH] Added lion-pytorch dependency to setup.py --- setup.py | 1 + 1 file changed, 1 insertion(+) diff --git a/setup.py b/setup.py index 7915abe..632e784 100644 --- 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",