We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 589b385 commit 629b0d2Copy full SHA for 629b0d2
training/eval.py
@@ -47,7 +47,7 @@ def wer(asr_model_name_or_path, prompts, audios, device, per_device_eval_batch_s
47
normalized_references = []
48
49
for pred, ref in zip(transcriptions, prompts):
50
- normalizer = english_normalizer if hasattr(pred, "language") and pred["language"] == "english" else basic_normalizer
+ normalizer = english_normalizer if return_language and pred["chunks"][0]["language"] == "english" else basic_normalizer
51
norm_ref = normalizer(ref)
52
if len(norm_ref) > 0:
53
norm_pred = normalizer(pred["text"])
0 commit comments