Fixed missing wandb dependency.

ZeroCool940711 · Aug 19, 2023 · 44f63f4 · 44f63f4
1 parent 9f52693
commit 44f63f4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -48,6 +48,7 @@
         "lion-pytorch",
         "omegaconf",
         "xformers>=0.0.20",
+        "wandb",
     ],
     classifiers=[
         "Development Status :: 4 - Beta",