Update model.py

model.py

@@ -3,24 +3,6 @@ from typing import List, Literal, Optional
 import types
 
 """Custom config to support modification of the Whisper encoder."""
-
-class ConvLayerConfig:
-    """Configuration for a single convolutional layer"""
-    def __init__(
-        self,
-        in_channels: int,
-        out_channels: int,
-        kernel_size: int,
-        stride: int = 1,
-        padding: int = 0,
-        activation: Literal["gelu", "relu", "none"] = "gelu"
-    ):
-        self.in_channels = in_channels
-        self.out_channels = out_channels
-        self.kernel_size = kernel_size
-        self.stride = stride
-        self.padding = padding
-        self.activation = activation
     
 class CustomWhisperConfig(WhisperConfig):
     def __init__(
@@ -43,19 +25,22 @@ class CustomWhisperConfig(WhisperConfig):
         
         if conv_preprocessing_layers is None:
             conv_preprocessing_layers = [
-                ConvLayerConfig(
-                    in_channels=self.num_mel_bins,
-                    out_channels=self.d_model,
-                    kernel_size=3,
-                    padding=1
-                ),
-                ConvLayerConfig(
-                    in_channels=self.d_model,
-                    out_channels=self.d_model,
-                    kernel_size=3,
-                    stride=2,
-                    padding=1
-                )
+                {
+                    "in_channels": self.num_mel_bins,
+                    "out_channels": self.d_model,
+                    "kernel_size": 3,
+                    "stride": 1,
+                    "padding": 1,
+                    "activation": "gelu"
+                },
+                {
+                    "in_channels": self.d_model,
+                    "out_channels": self.d_model,
+                    "kernel_size": 3,
+                    "stride": 2,
+                    "padding": 1,
+                    "activation": "gelu"
+                }
             ]
             
         self.conv_preprocessing_layers = conv_preprocessing_layers