diff --git a/requirements.txt b/requirements.txt
index eda3109..33d8673 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,3 +2,4 @@ torch==1.9.0
 torchvision==0.10.0
 pillow==8.3.1
 numpy==1.19.5
+timm==0.5.4