Skip to content

Commit 09a3e08

Browse files
authored
remove unnecessary entries in some auto model mappings (#45224)
fixxxxxx Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
1 parent df557a2 commit 09a3e08

2 files changed

Lines changed: 2 additions & 4 deletions

File tree

src/transformers/models/auto/modeling_auto.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1009,7 +1009,6 @@ class _BaseModelWithGenerate(PreTrainedModel, GenerationMixin):
10091009
("perception_lm", "PerceptionLMForConditionalGeneration"),
10101010
("pi0", "PI0ForConditionalGeneration"),
10111011
("pix2struct", "Pix2StructForConditionalGeneration"),
1012-
("pixtral", "LlavaForConditionalGeneration"),
10131012
("pp_chart2table", "GotOcr2ForConditionalGeneration"),
10141013
("qwen2_5_vl", "Qwen2_5_VLForConditionalGeneration"),
10151014
("qwen2_vl", "Qwen2VLForConditionalGeneration"),
@@ -1671,7 +1670,6 @@ class _BaseModelWithGenerate(PreTrainedModel, GenerationMixin):
16711670
# Model for Text-To-Waveform mapping
16721671
("bark", "BarkModel"),
16731672
("csm", "CsmForConditionalGeneration"),
1674-
("fastspeech2_conformer", "FastSpeech2ConformerWithHifiGan"),
16751673
("fastspeech2_conformer_with_hifigan", "FastSpeech2ConformerWithHifiGan"),
16761674
("higgs_audio_v2", "HiggsAudioV2ForConditionalGeneration"),
16771675
("musicgen", "MusicgenForConditionalGeneration"),

src/transformers/models/fastspeech2_conformer/configuration_fastspeech2_conformer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -262,7 +262,7 @@ def validate_architecture(self):
262262
raise ValueError("Either use_masking or use_weighted_masking can be True, but not both.")
263263

264264

265-
@auto_docstring(checkpoint="espnet/fastspeech2_conformer")
265+
@auto_docstring(checkpoint="espnet/fastspeech2_conformer_with_hifigan")
266266
@strict
267267
class FastSpeech2ConformerHifiGanConfig(PreTrainedConfig):
268268
r"""
@@ -323,7 +323,7 @@ def __post_init__(self, **kwargs):
323323
super().__post_init__(**kwargs)
324324

325325

326-
@auto_docstring(checkpoint="espnet/fastspeech2_conformer")
326+
@auto_docstring(checkpoint="espnet/fastspeech2_conformer_with_hifigan")
327327
@strict
328328
class FastSpeech2ConformerWithHifiGanConfig(PreTrainedConfig):
329329
r"""

0 commit comments

Comments
 (0)