flip these two

Author: YassineYousfi

timm/models/fastvit.py

@@ -421,8 +421,8 @@ def _fuse_bn(
 def convolutional_stem(
         in_chs: int,
         out_chs: int,
-        inference_mode: bool = False,
         act_layer: nn.Module = nn.GELU,
+        inference_mode: bool = False
 ) -> nn.Sequential:
     """Build convolutional stem with MobileOne blocks.
 
@@ -1124,8 +1124,8 @@ def __init__(
         self.stem = convolutional_stem(
             in_chans,
             embed_dims[0],
+            act_layer,
             inference_mode,
-            act_layer
         )
 
         # Build the main stages of the network architecture
@@ -1187,6 +1187,7 @@ def __init__(
                 group_size=1,
                 inference_mode=inference_mode,
                 use_se=True,
+                act_layer=act_layer,
                 num_conv_branches=1,
             )
             self.head = ClassifierHead(