diff --git a/setup.py b/setup.py
index de927de..dbaf0fe 100644
--- 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",
     ],