diff --git a/modeling_intern_vit.py b/modeling_intern_vit.py
index 233b029..eb4aef4 100644
--- a/modeling_intern_vit.py
+++ b/modeling_intern_vit.py
@@ -10,7 +10,7 @@ import torch
 import torch.nn.functional as F
 import torch.utils.checkpoint
 from einops import rearrange
-from timm.models.layers import DropPath
+from timm.layers import DropPath
 from torch import nn
 from transformers.activations import ACT2FN
 from transformers.modeling_outputs import (BaseModelOutput,