Update encoder.py

encoder.py

@@ -515,7 +515,7 @@ class ConformerEncoder(nn.Module):
             )
             self.layers.append(layer)
 
-        self.pos_enc.extend_pe(pos_emb_max_len, next(self.parameters()).device)
+        self.pos_enc.extend_pe(pos_emb_max_len)
 
     def input_example(
         self,