diff --git a/src/transformers/models/auto/modeling_auto.py b/src/transformers/models/auto/modeling_auto.py index f7228c078e9e..44d83634d28e 100644 --- a/src/transformers/models/auto/modeling_auto.py +++ b/src/transformers/models/auto/modeling_auto.py @@ -1009,10 +1009,12 @@ class _BaseModelWithGenerate(PreTrainedModel, GenerationMixin): ("pi0", "PI0ForConditionalGeneration"), ("pix2struct", "Pix2StructForConditionalGeneration"), ("pp_chart2table", "GotOcr2ForConditionalGeneration"), + ("qwen2_5_omni_thinker", "Qwen2_5OmniThinkerForConditionalGeneration"), ("qwen2_5_vl", "Qwen2_5_VLForConditionalGeneration"), ("qwen2_vl", "Qwen2VLForConditionalGeneration"), ("qwen3_5", "Qwen3_5ForConditionalGeneration"), ("qwen3_5_moe", "Qwen3_5MoeForConditionalGeneration"), + ("qwen3_omni_moe_thinker", "Qwen3OmniMoeThinkerForConditionalGeneration"), ("qwen3_vl", "Qwen3VLForConditionalGeneration"), ("qwen3_vl_moe", "Qwen3VLMoeForConditionalGeneration"), ("shieldgemma2", "Gemma3ForConditionalGeneration"), diff --git a/utils/check_repo.py b/utils/check_repo.py index 9c7fb8714dbe..c4b8e44b4dd8 100644 --- a/utils/check_repo.py +++ b/utils/check_repo.py @@ -470,7 +470,6 @@ "PaddleOCRTextModel", # Building part of bigger (tested) model "Qwen2_5OmniTalkerForConditionalGeneration", # Building part of a bigger model "Qwen2_5OmniTalkerModel", # Building part of a bigger model - "Qwen2_5OmniThinkerForConditionalGeneration", # Building part of a bigger model "Qwen2_5OmniThinkerTextModel", # Building part of a bigger model "Qwen2_5OmniToken2WavModel", # Building part of a bigger model "Qwen2_5OmniToken2WavBigVGANModel", # Building part of a bigger model @@ -488,7 +487,6 @@ "Qwen3OmniMoeTalkerCodePredictorModelForConditionalGeneration", # Building part of a bigger model "Qwen3OmniMoeTalkerForConditionalGeneration", # Building part of a bigger model "Qwen3OmniMoeTalkerModel", # Building part of a bigger model - "Qwen3OmniMoeThinkerForConditionalGeneration", # Building part of a bigger model "Qwen3OmniMoeThinkerTextModel", # Building part of a bigger model "Ernie4_5_VLMoeTextModel", # Building part of a bigger model "PeAudioFrameLevelModel",