Update test.py (#560)

2024-01-29 17:30:44 +08:00
parent be84932f86
commit 0aa47e5ccc
1 changed files with 1 additions and 1 deletions
--- a/scripts/whisper/test.py
+++ b/scripts/whisper/test.py
@@ -259,7 +259,7 @@ def compute_features(filename: str) -> torch.Tensor:

    # We pad 50 frames at the end so that it is able to detect eot
    # You can use another value instead of 50.
-    mel = torch.nn.functional.pad(mel, (0, 0, 0, 50), "constant", 0)
+    mel = torch.nn.functional.pad(mel, (0, 0, 0, 1000), "constant", 0)
    # Note that if it throws for a multilingual model,
    # please use a larger value, say 300