fix: Add generation_max_length to Whisper configs

alexandrainst · Oct 25, 2023 · 52a0860 · 52a0860
1 parent 27750d1
commit 52a0860
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 0 deletions.
diff --git a/config/model/test_whisper.yaml b/config/model/test_whisper.yaml
@@ -15,3 +15,4 @@ mask_time_prob: 0.5
 mask_time_length: 10
 mask_feature_prob: 0.5
 mask_feature_length: 64
+generation_max_length: 128
diff --git a/config/model/whisper_large.yaml b/config/model/whisper_large.yaml
@@ -15,3 +15,4 @@ mask_time_prob: 0.5
 mask_time_length: 10
 mask_feature_prob: 0.5
 mask_feature_length: 64
+generation_max_length: 128
diff --git a/config/model/whisper_medium.yaml b/config/model/whisper_medium.yaml
@@ -15,3 +15,4 @@ mask_time_prob: 0.5
 mask_time_length: 10
 mask_feature_prob: 0.5
 mask_feature_length: 64
+generation_max_length: 128
diff --git a/config/model/whisper_small.yaml b/config/model/whisper_small.yaml
@@ -15,3 +15,4 @@ mask_time_prob: 0.5
 mask_time_length: 10
 mask_feature_prob: 0.5
 mask_feature_length: 64
+generation_max_length: 128
diff --git a/config/model/whisper_xsmall.yaml b/config/model/whisper_xsmall.yaml
@@ -15,3 +15,4 @@ mask_time_prob: 0.5
 mask_time_length: 10
 mask_feature_prob: 0.5
 mask_feature_length: 64
+generation_max_length: 128
diff --git a/config/model/whisper_xxsmall.yaml b/config/model/whisper_xxsmall.yaml
@@ -15,3 +15,4 @@ mask_time_prob: 0.5
 mask_time_length: 10
 mask_feature_prob: 0.5
 mask_feature_length: 64
+generation_max_length: 128