Update BEN2.py

BEN2.py

@@ -6,7 +6,7 @@ import torch.nn.functional as F
 from einops import rearrange
 import torch.utils.checkpoint as checkpoint
 import numpy as np
-from timm.layers import DropPath, to_2tuple, trunc_normal_
+from timm.models.layers  import DropPath, to_2tuple, trunc_normal_
 from PIL import Image, ImageOps
 from torchvision import transforms
 import numpy as np