[fix] Add FastSpeech2ConformerWithHifiGan (#38207)

* add to mapping

* oops

* oops

* add to config_mapping_names

* revert

* fix?

* config-mapping-names

* fix?

* fix?
This commit is contained in:
Steven Liu 2025-06-27 09:38:21 -07:00 committed by GitHub
parent c2dc72bb5f
commit 02a769b058
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 7 additions and 1 deletions

View File

@ -130,6 +130,7 @@ CONFIG_MAPPING_NAMES = OrderedDict[str, str](
("falcon_h1", "FalconH1Config"), ("falcon_h1", "FalconH1Config"),
("falcon_mamba", "FalconMambaConfig"), ("falcon_mamba", "FalconMambaConfig"),
("fastspeech2_conformer", "FastSpeech2ConformerConfig"), ("fastspeech2_conformer", "FastSpeech2ConformerConfig"),
("fastspeech2_conformer_with_hifigan", "FastSpeech2ConformerWithHifiGanConfig"),
("flaubert", "FlaubertConfig"), ("flaubert", "FlaubertConfig"),
("flava", "FlavaConfig"), ("flava", "FlavaConfig"),
("fnet", "FNetConfig"), ("fnet", "FNetConfig"),
@ -511,6 +512,7 @@ MODEL_NAMES_MAPPING = OrderedDict[str, str](
("falcon_h1", "FalconH1"), ("falcon_h1", "FalconH1"),
("falcon_mamba", "FalconMamba"), ("falcon_mamba", "FalconMamba"),
("fastspeech2_conformer", "FastSpeech2Conformer"), ("fastspeech2_conformer", "FastSpeech2Conformer"),
("fastspeech2_conformer_with_hifigan", "FastSpeech2ConformerWithHifiGan"),
("flan-t5", "FLAN-T5"), ("flan-t5", "FLAN-T5"),
("flan-ul2", "FLAN-UL2"), ("flan-ul2", "FLAN-UL2"),
("flaubert", "FlauBERT"), ("flaubert", "FlauBERT"),
@ -866,6 +868,7 @@ SPECIAL_MODEL_TYPE_TO_MODULE_NAME = OrderedDict[str, str](
("sam_hq_vision_model", "sam_hq"), ("sam_hq_vision_model", "sam_hq"),
("llama4_text", "llama4"), ("llama4_text", "llama4"),
("blip_2_qformer", "blip_2"), ("blip_2_qformer", "blip_2"),
("fastspeech2_conformer_with_hifigan", "fastspeech2_conformer"),
] ]
) )
@ -1178,7 +1181,8 @@ class AutoConfig:
>>> unused_kwargs >>> unused_kwargs
{'foo': False} {'foo': False}
```""" ```
"""
use_auth_token = kwargs.pop("use_auth_token", None) use_auth_token = kwargs.pop("use_auth_token", None)
if use_auth_token is not None: if use_auth_token is not None:
warnings.warn( warnings.warn(

View File

@ -121,6 +121,7 @@ MODEL_MAPPING_NAMES = OrderedDict(
("falcon_h1", "FalconH1Model"), ("falcon_h1", "FalconH1Model"),
("falcon_mamba", "FalconMambaModel"), ("falcon_mamba", "FalconMambaModel"),
("fastspeech2_conformer", "FastSpeech2ConformerModel"), ("fastspeech2_conformer", "FastSpeech2ConformerModel"),
("fastspeech2_conformer_with_hifigan", "FastSpeech2ConformerWithHifiGan"),
("flaubert", "FlaubertModel"), ("flaubert", "FlaubertModel"),
("flava", "FlavaModel"), ("flava", "FlavaModel"),
("fnet", "FNetModel"), ("fnet", "FNetModel"),
@ -1512,6 +1513,7 @@ MODEL_FOR_TEXT_TO_WAVEFORM_MAPPING_NAMES = OrderedDict(
("bark", "BarkModel"), ("bark", "BarkModel"),
("csm", "CsmForConditionalGeneration"), ("csm", "CsmForConditionalGeneration"),
("fastspeech2_conformer", "FastSpeech2ConformerWithHifiGan"), ("fastspeech2_conformer", "FastSpeech2ConformerWithHifiGan"),
("fastspeech2_conformer_with_hifigan", "FastSpeech2ConformerWithHifiGan"),
("musicgen", "MusicgenForConditionalGeneration"), ("musicgen", "MusicgenForConditionalGeneration"),
("musicgen_melody", "MusicgenMelodyForConditionalGeneration"), ("musicgen_melody", "MusicgenMelodyForConditionalGeneration"),
("qwen2_5_omni", "Qwen2_5OmniForConditionalGeneration"), ("qwen2_5_omni", "Qwen2_5OmniForConditionalGeneration"),