diff --git a/modules/BLIP/blip_vit.py b/modules/BLIP/blip_vit.py
index cec3d8e..c1aabac 100644
--- a/modules/BLIP/blip_vit.py
+++ b/modules/BLIP/blip_vit.py
@@ -15,7 +15,7 @@ from functools import partial
 
 from timm.models.vision_transformer import _cfg, PatchEmbed
 from timm.models.registry import register_model
-from timm.models.layers import trunc_normal_, DropPath
+from timm.layers import trunc_normal_, DropPath
 from timm.models.helpers import named_apply, adapt_input_conv
 
 from fairscale.nn.checkpoint.checkpoint_activations import checkpoint_wrapper