Mirroristas/whisper
2
0
Fork 0
mirror of https://github.com/openai/whisper.git synced 2025-03-30 14:28:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

need to pass in cut region to audio encoder too

Browse Source
This commit is contained in:
Amal Jacob 2025-02-07 13:42:03 -08:00
parent c29813b560
commit 97bbe4b70f
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
whisper/model.py
View File

@ -206,7 +206,7 @@ class AudioEncoderTokenPruner():
class AudioEncoder(nn.Module):
def __init__(
self, n_mels: int, n_ctx: int, n_state: int, n_head: int, n_layer: int, ext_feat_flag: bool = False, cut_region: Tuple[int, int]=[750, 1000]
self, n_mels: int, n_ctx: int, n_state: int, n_head: int, n_layer: int, ext_feat_flag: bool = False, cut_region: Tuple[int, int]=None
):
super().__init__()
self.conv1 = Conv1d(n_mels, n_state, kernel_size=3, padding=1)
@ -301,6 +301,7 @@ class Whisper(nn.Module):
self.dims.n_audio_head,
self.dims.n_audio_layer,
ext_feat_flag=ext_feat_flag,
cut_region=cut_region,
)
self.decoder = TextDecoder(
self.dims.n_vocab,