fix import from timm.models.layers warning

py/BiRefNet_legacy/backbones/pvt_v2.py

@@ -2,7 +2,7 @@
 import torch.nn as nn
 from functools import partial
 
-from timm.models.layers import DropPath, to_2tuple, trunc_normal_
+from timm.layers import DropPath, to_2tuple, trunc_normal_
 from timm.models import register_model
 
 import math

py/BiRefNet_legacy/backbones/swin_v1.py

@@ -10,7 +10,7 @@
 import torch.nn.functional as F
 import torch.utils.checkpoint as checkpoint
 import numpy as np
-from timm.models.layers import DropPath, to_2tuple, trunc_normal_
+from timm.layers import DropPath, to_2tuple, trunc_normal_
 
 from ..config import Config
 

py/BiRefNet_legacy/modules/mlp.py

@@ -2,7 +2,7 @@
 import torch.nn as nn
 from functools import partial
 
-from timm.models.layers import DropPath, to_2tuple, trunc_normal_
+from timm.layers import DropPath, to_2tuple, trunc_normal_
 from timm.models import register_model
 
 import math

py/BiRefNet_v2/models/backbones/pvt_v2.py

@@ -2,7 +2,7 @@
 import torch.nn as nn
 from functools import partial
 
-from timm.models.layers import DropPath, to_2tuple, trunc_normal_
+from timm.layers import DropPath, to_2tuple, trunc_normal_
 from timm.models import register_model
 
 import math

py/BiRefNet_v2/models/backbones/swin_v1.py

@@ -10,7 +10,7 @@
 import torch.nn.functional as F
 import torch.utils.checkpoint as checkpoint
 import numpy as np
-from timm.models.layers import DropPath, to_2tuple, trunc_normal_
+from timm.layers import DropPath, to_2tuple, trunc_normal_
 
 from ...config import Config
 

py/BiRefNet_v2/models/modules/mlp.py

@@ -2,7 +2,7 @@
 import torch.nn as nn
 from functools import partial
 
-from timm.models.layers import DropPath, to_2tuple, trunc_normal_
+from timm.layers import DropPath, to_2tuple, trunc_normal_
 from timm.models import register_model
 
 import math

py/ben_ultra.py

@@ -7,7 +7,7 @@
 import torch.utils.checkpoint as checkpoint
 from einops import rearrange
 from PIL import Image, ImageFilter, ImageOps
-from timm.models.layers import DropPath, to_2tuple, trunc_normal_
+from timm.layers import DropPath, to_2tuple, trunc_normal_
 from torchvision import transforms
 
 

py/evf_sam/model/unilm/beit3/modeling_utils.py

@@ -8,7 +8,7 @@
 import math
 import torch
 import torch.nn as nn
-from timm.models.layers import trunc_normal_ as __call_trunc_normal_
+from timm.layers import trunc_normal_ as __call_trunc_normal_
 
 from torchscale.model.BEiT3 import BEiT3
 from torchscale.architecture.config import EncoderConfig

py/local_groundingdino/models/GroundingDINO/backbone/swin_transformer.py

@@ -16,7 +16,7 @@
 import torch.nn as nn
 import torch.nn.functional as F
 import torch.utils.checkpoint as checkpoint
-from timm.models.layers import DropPath, to_2tuple, trunc_normal_
+from timm.layers import DropPath, to_2tuple, trunc_normal_
 
 from local_groundingdino.util.misc import NestedTensor
 

py/local_groundingdino/models/GroundingDINO/fuse_modules.py

@@ -8,7 +8,7 @@
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
-from timm.models.layers import DropPath
+from timm.layers import DropPath
 
 
 class FeatureResizer(nn.Module):

py/sam_hq/modeling/tiny_vit.py

@@ -12,7 +12,7 @@
 import torch.nn as nn
 import torch.nn.functional as F
 import torch.utils.checkpoint as checkpoint
-from timm.models.layers import DropPath as TimmDropPath,\
+from timm.layers import DropPath as TimmDropPath,\
     to_2tuple, trunc_normal_
 from timm.models import register_model
 from typing import Tuple