From 4e682f1fda674f9dd5bc8afc662eec3831ada8da Mon Sep 17 00:00:00 2001
From: Fredo Guan <fredo.guan@hotmail.com>
Date: Sun, 8 Jan 2023 11:52:13 -0800
Subject: [PATCH] Update metaformers.py

---
 timm/models/metaformers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/timm/models/metaformers.py b/timm/models/metaformers.py
index 9bb85c44..e694fd67 100644
--- a/timm/models/metaformers.py
+++ b/timm/models/metaformers.py
@@ -228,7 +228,7 @@ class Downsampling(nn.Module):
     """
     def __init__(self, in_channels, out_channels, 
         kernel_size, stride=1, padding=0, 
-        pre_norm=None, post_norm=None):
+        pre_norm=None, post_norm=None, pre_permute = False):
         super().__init__()
         self.pre_norm = pre_norm(in_channels) if pre_norm else nn.Identity()
         self.pre_permute = pre_permute