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

nit

Browse Source
This commit is contained in:
Amal Jacob 2025-01-26 21:22:34 -08:00
parent 9d8d372a4d
commit 34db988568
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
whisper/model.py
View File

@ -207,6 +207,7 @@ class AudioEncoder(nn.Module):
for block in self.blocks: for block in self.blocks:
x = block(x) x = block(x)
if block.attn.attention_scores is not None: if block.attn.attention_scores is not None:
print(f"Captured attention scores from layer {len(self.all_attention_scores)}")
self.all_attention_scores.append(block.attn.attention_scores) self.all_attention_scores.append(block.attn.attention_scores)
x = self.ln_post(x) x = self.ln_post(x)