Add a few missing __all__ entries.

timm/models/efficientformer_v2.py

@@ -29,6 +29,8 @@ from ._manipulate import checkpoint_seq
 from ._registry import generate_default_cfgs, register_model
 
 
+__all__ = ['EfficientFormerV2']
+
 EfficientFormer_width = {
     'L': (40, 80, 192, 384),  # 26m 83.3% 6attn
     'S2': (32, 64, 144, 288),  # 12m 81.6% 4attn dp0.02

timm/models/fastvit.py

@@ -20,6 +20,7 @@ from ._features import feature_take_indices
 from ._manipulate import checkpoint_seq
 from ._registry import register_model, generate_default_cfgs
 
+__all__ = ['FastVit']
 
 def num_groups(group_size, channels):
     if not group_size:  # 0 or None

timm/models/hiera.py

@@ -42,6 +42,9 @@ from ._features import feature_take_indices
 from ._features_fx import register_notrace_function
 
 
+__all__ = ['Hiera']
+
+
 def conv_nd(n: int) -> Type[nn.Module]:
     """
     Returns a conv with nd (e.g., Conv2d for n=2). Work up to n=3.

timm/models/nextvit.py

@@ -20,6 +20,8 @@ from ._features_fx import register_notrace_function
 from ._manipulate import checkpoint_seq
 from ._registry import generate_default_cfgs, register_model
 
+__all__ = ['NextViT']
+
 
 def merge_pre_bn(module, pre_bn_1, pre_bn_2=None):
     """ Merge pre BN to reduce inference runtime.