diff --git a/funasr/download/name_maps_from_hub.py b/funasr/download/name_maps_from_hub.py index 12b66d3e9..a18859877 100644 --- a/funasr/download/name_maps_from_hub.py +++ b/funasr/download/name_maps_from_hub.py @@ -21,7 +21,7 @@ name_maps_ms = { name_maps_hf = { "paraformer": "funasr/paraformer-zh", "paraformer-zh": "funasr/paraformer-zh", - "paraformer-en": "funasr/paraformer-zh", + "paraformer-en": "funasr/paraformer-en", "paraformer-zh-streaming": "funasr/paraformer-zh-streaming", "fsmn-vad": "funasr/fsmn-vad", "ct-punc": "funasr/ct-punc", diff --git a/tests/test_hf_model_aliases.py b/tests/test_hf_model_aliases.py new file mode 100644 index 000000000..0c75af0a9 --- /dev/null +++ b/tests/test_hf_model_aliases.py @@ -0,0 +1,5 @@ +from funasr.download.name_maps_from_hub import name_maps_hf + + +def test_hf_paraformer_en_alias_targets_english_checkpoint(): + assert name_maps_hf["paraformer-en"] == "funasr/paraformer-en"