Make Swin and Swinv2 untouched

huggingface · Aug 8, 2022 · 2a30ae5 · 2a30ae5
1 parent 9cbb1b8
commit 2a30ae5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/src/transformers/models/swin/modeling_swin.py b/src/transformers/models/swin/modeling_swin.py
@@ -985,8 +985,7 @@ def forward(
         )
 
         sequence_output = encoder_outputs[0]
-        if self.layernorm is not None:
-            sequence_output = self.layernorm(sequence_output)
+        sequence_output = self.layernorm(sequence_output)
 
         pooled_output = None
         if self.pooler is not None:

diff --git a/src/transformers/models/swinv2/modeling_swinv2.py b/src/transformers/models/swinv2/modeling_swinv2.py
@@ -1065,8 +1065,7 @@ def forward(
         )
 
         sequence_output = encoder_outputs[0]
-        if self.layernorm is not None:
-            sequence_output = self.layernorm(sequence_output)
+        sequence_output = self.layernorm(sequence_output)
 
         pooled_output = None
         if self.pooler is not None: