diff --git a/nemo/collections/common/prompts/canary.py b/nemo/collections/common/prompts/canary.py index e31ccf84c99e..d62de26a8881 100644 --- a/nemo/collections/common/prompts/canary.py +++ b/nemo/collections/common/prompts/canary.py @@ -16,7 +16,9 @@ class CanaryPromptFormatter(PromptFormatter): "template": f"{CANARY_BOS}|source_lang||task||target_lang||pnc|", "slots": { "source_lang": Modality.Text, - "task": Modality.TextLiteral("asr", "ast", "translate", "transcribe", "s2t_translation", "<|transcribe|>", "<|translate|>"), + "task": Modality.TextLiteral( + "asr", "ast", "translate", "transcribe", "s2t_translation", "<|transcribe|>", "<|translate|>" + ), "target_lang": Modality.Text, "pnc": Modality.TextLiteral("yes", "no", "<|pnc|>", "<|nopnc|>"), },