diff --git a/requirements.txt b/requirements.txt
index 6e8108d..c99c6be 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,3 +5,4 @@ transformers
 av
 tensorboard
 colossalai
+timm