Update metaformers.py

timm/models/metaformers.py

@@ -650,7 +650,7 @@ class MetaFormer(nn.Module):
         stages = nn.ModuleList() # each stage consists of multiple metaformer blocks
         cur = 0
         for i in range(num_stage):
-            stage = nn.Sequential(OrderedDict[
+            stage = nn.Sequential(OrderedDict([
                 ('downsample', downsample_layers[i]),
                 ('blocks', nn.Sequential(*[MetaFormerBlock(
                     dim=dims[i],
@@ -661,7 +661,7 @@ class MetaFormer(nn.Module):
                     layer_scale_init_value=layer_scale_init_values[i],
                     res_scale_init_value=res_scale_init_values[i]
                     ) for j in range(depths[i])])
-                )]
+                )])
             )
             stages.append(stage)
             cur += depths[i]