Update setup.py

ZeroCool940711 · Jun 4, 2023 · 034402a · 034402a
1 parent e1b7f93
commit 034402a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -35,6 +35,8 @@
         "torchvision",
         "torch_optimizer",
         "tqdm",
+        "timm",
+        "tqdm-loggable",
         "vector-quantize-pytorch>=0.10.14",
         "lion-pytorch",
     ],