From 034402ab152ce89d0416709f417581c9e45b7e33 Mon Sep 17 00:00:00 2001
From: Korakoe <56580073+korakoe@users.noreply.github.com>
Date: Sun, 4 Jun 2023 15:11:31 +0800
Subject: [PATCH] Update setup.py

---
 setup.py | 2 ++
 1 file changed, 2 insertions(+)

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",
     ],