Update cvt.py

Author: fffffgggg54

timm/models/cvt.py

@@ -314,6 +314,7 @@ def __init__(
         embed_kernel_size: int = 7,
         embed_stride: int = 4,
         embed_padding: int = 2,
+        embed_norm_layer: nn.Module = partial(LayerNorm2d, eps=1e-5),
         kernel_size: int = 3,
         stride_q: int = 1,
         stride_kv: int = 2,
@@ -405,6 +406,7 @@ def __init__(
         embed_kernel_size: Tuple[int, ...] = (7, 3, 3),
         embed_stride: Tuple[int, ...] = (4, 2, 2),
         embed_padding: Tuple[int, ...] = (2, 1, 1),
+        embed_norm_layer: nn.Module = partial(LayerNorm2d, eps=1e-5),
         kernel_size: int = 3,
         stride_q: int = 1,
         stride_kv: int = 2,
@@ -452,6 +454,7 @@ def __init__(
                 embed_kernel_size = embed_kernel_size[stage_idx],
                 embed_stride = embed_stride[stage_idx],
                 embed_padding = embed_padding[stage_idx],
+                embed_norm_layer = embed_norm_layer,
                 kernel_size = kernel_size,
                 stride_q = stride_q,
                 stride_kv = stride_kv,