diff --git a/py/BiRefNet_legacy/backbones/pvt_v2.py b/py/BiRefNet_legacy/backbones/pvt_v2.py index 7164f85..d5b7005 100644 --- a/py/BiRefNet_legacy/backbones/pvt_v2.py +++ b/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 diff --git a/py/BiRefNet_legacy/backbones/swin_v1.py b/py/BiRefNet_legacy/backbones/swin_v1.py index 57501dc..f6cec34 100644 --- a/py/BiRefNet_legacy/backbones/swin_v1.py +++ b/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 diff --git a/py/BiRefNet_legacy/modules/mlp.py b/py/BiRefNet_legacy/modules/mlp.py index a383459..7aa0138 100644 --- a/py/BiRefNet_legacy/modules/mlp.py +++ b/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 diff --git a/py/BiRefNet_v2/models/backbones/pvt_v2.py b/py/BiRefNet_v2/models/backbones/pvt_v2.py index 4b902dd..0d9d818 100644 --- a/py/BiRefNet_v2/models/backbones/pvt_v2.py +++ b/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 diff --git a/py/BiRefNet_v2/models/backbones/swin_v1.py b/py/BiRefNet_v2/models/backbones/swin_v1.py index 7739622..e116716 100644 --- a/py/BiRefNet_v2/models/backbones/swin_v1.py +++ b/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 diff --git a/py/BiRefNet_v2/models/modules/mlp.py b/py/BiRefNet_v2/models/modules/mlp.py index a383459..7aa0138 100644 --- a/py/BiRefNet_v2/models/modules/mlp.py +++ b/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 diff --git a/py/ben_ultra.py b/py/ben_ultra.py index e5222af..5ec3399 100644 --- a/py/ben_ultra.py +++ b/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 diff --git a/py/evf_sam/model/unilm/beit3/modeling_utils.py b/py/evf_sam/model/unilm/beit3/modeling_utils.py index 6581159..f7c6e07 100644 --- a/py/evf_sam/model/unilm/beit3/modeling_utils.py +++ b/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 diff --git a/py/local_groundingdino/models/GroundingDINO/backbone/swin_transformer.py b/py/local_groundingdino/models/GroundingDINO/backbone/swin_transformer.py index c35b5c2..5261c1a 100644 --- a/py/local_groundingdino/models/GroundingDINO/backbone/swin_transformer.py +++ b/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 diff --git a/py/local_groundingdino/models/GroundingDINO/fuse_modules.py b/py/local_groundingdino/models/GroundingDINO/fuse_modules.py index 2753b3d..a5d428c 100644 --- a/py/local_groundingdino/models/GroundingDINO/fuse_modules.py +++ b/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): diff --git a/py/sam_hq/modeling/tiny_vit.py b/py/sam_hq/modeling/tiny_vit.py index 38b7bf9..03a9fdb 100644 --- a/py/sam_hq/modeling/tiny_vit.py +++ b/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