diff --git a/requirements.txt b/requirements.txt
index 56cc61764..16e8d74d8 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,7 +8,7 @@ scikit-image>=0.19.3
 scikit-learn>=1.0.2
 scipy>=1.10.0
 tensorboard>=2.9.1
-timm>=0.5.4
+timm>=1.0.11
 torch>=1.12.0
 torchaudio>=0.12.0
 torchvision>=0.13.0
diff --git a/semilearn/nets/vit/vit.py b/semilearn/nets/vit/vit.py
index 84b5aa327..333e7a23c 100644
--- a/semilearn/nets/vit/vit.py
+++ b/semilearn/nets/vit/vit.py
@@ -9,8 +9,8 @@
 import torch.nn.functional as F
 import torch.utils.checkpoint
 
-from timm.models.layers import DropPath, trunc_normal_
-from timm.models.layers.helpers import to_2tuple
+from timm.layers import DropPath, trunc_normal_
+from timm.layers.helpers import to_2tuple
 
 from semilearn.nets.utils import load_checkpoint