diff --git a/timm/models/metaformers.py b/timm/models/metaformers.py index 92d7a0c5..a24e3377 100644 --- a/timm/models/metaformers.py +++ b/timm/models/metaformers.py @@ -21,7 +21,7 @@ from functools import partial import torch import torch.nn as nn from timm.layers import trunc_normal_, DropPath -from timm.models.registry import register_model +from ._registry import register_model from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD from timm.layers.helpers import to_2tuple