Skip to content

Commit 8b8c576

Browse files
Merge pull request #61 from sanchit-gandhi/eval-fix
[eval] fix trigger for english normaliser
2 parents 589b385 + 629b0d2 commit 8b8c576

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

training/eval.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ def wer(asr_model_name_or_path, prompts, audios, device, per_device_eval_batch_s
4747
normalized_references = []
4848

4949
for pred, ref in zip(transcriptions, prompts):
50-
normalizer = english_normalizer if hasattr(pred, "language") and pred["language"] == "english" else basic_normalizer
50+
normalizer = english_normalizer if return_language and pred["chunks"][0]["language"] == "english" else basic_normalizer
5151
norm_ref = normalizer(ref)
5252
if len(norm_ref) > 0:
5353
norm_pred = normalizer(pred["text"])

0 commit comments

Comments
 (0)