fix #4

model.py

@@ -1032,10 +1032,9 @@ class WhisperEncoder(WhisperPreTrainedModel):
 
     # CUSTOM
     def get_conv_stride(self):
-        """Calculate total stride of all conv layers"""
         total_stride = 1
         for layer in self.conv_layers:
-            total_stride *= layer.stride[0]
+            total_stride *= layer[0].stride[0]
         return total_stride
 
     def _freeze_parameters(self):
@@ -1116,7 +1115,7 @@ class WhisperEncoder(WhisperPreTrainedModel):
         """
 
         # CUSTOM
-        expected_seq_length = self.config.max_source_positions * self.get_conv_stride()
+        #expected_seq_length = self.config.max_source_positions * self.get_conv_stride()
         
         # CUSTOM
         # Must be deactivated for our purpose, theoretically Whisper supports any sequence length for the encoder