transformers 5.0.0rc0__py3-none-any.whl → 5.0.0rc2__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- transformers/__init__.py +49 -3
- transformers/activations.py +1 -1
- transformers/audio_utils.py +0 -1
- transformers/cache_utils.py +17 -15
- transformers/cli/serve.py +47 -17
- transformers/configuration_utils.py +114 -70
- transformers/conversion_mapping.py +83 -7
- transformers/convert_slow_tokenizer.py +225 -10
- transformers/core_model_loading.py +374 -147
- transformers/data/data_collator.py +12 -4
- transformers/dependency_versions_table.py +2 -3
- transformers/dynamic_module_utils.py +1 -2
- transformers/feature_extraction_utils.py +55 -24
- transformers/file_utils.py +0 -1
- transformers/generation/__init__.py +11 -1
- transformers/generation/candidate_generator.py +79 -31
- transformers/generation/configuration_utils.py +165 -124
- transformers/generation/continuous_batching/__init__.py +4 -0
- transformers/generation/continuous_batching/cache.py +47 -18
- transformers/generation/continuous_batching/cache_manager.py +131 -34
- transformers/generation/continuous_batching/continuous_api.py +228 -136
- transformers/generation/continuous_batching/requests.py +28 -1
- transformers/generation/continuous_batching/scheduler.py +11 -4
- transformers/generation/stopping_criteria.py +1 -1
- transformers/generation/utils.py +108 -110
- transformers/generation/watermarking.py +8 -5
- transformers/image_processing_base.py +3 -14
- transformers/image_processing_utils_fast.py +15 -4
- transformers/initialization.py +37 -0
- transformers/integrations/__init__.py +16 -2
- transformers/integrations/accelerate.py +58 -113
- transformers/integrations/aqlm.py +36 -66
- transformers/integrations/awq.py +46 -515
- transformers/integrations/bitnet.py +47 -105
- transformers/integrations/bitsandbytes.py +91 -202
- transformers/integrations/deepspeed.py +18 -2
- transformers/integrations/eetq.py +84 -81
- transformers/integrations/fbgemm_fp8.py +191 -145
- transformers/integrations/finegrained_fp8.py +241 -208
- transformers/integrations/flash_attention.py +2 -2
- transformers/integrations/fp_quant.py +92 -0
- transformers/integrations/ggml.py +11 -1
- transformers/integrations/higgs.py +37 -62
- transformers/integrations/hub_kernels.py +65 -8
- transformers/integrations/integration_utils.py +45 -0
- transformers/integrations/mistral.py +12 -0
- transformers/integrations/moe.py +240 -0
- transformers/integrations/mxfp4.py +28 -74
- transformers/integrations/peft.py +12 -29
- transformers/integrations/quanto.py +77 -56
- transformers/integrations/quark.py +55 -0
- transformers/integrations/spqr.py +42 -90
- transformers/integrations/tensor_parallel.py +167 -221
- transformers/integrations/torchao.py +32 -38
- transformers/integrations/vptq.py +40 -59
- transformers/modelcard.py +1 -2
- transformers/modeling_gguf_pytorch_utils.py +74 -19
- transformers/modeling_rope_utils.py +107 -86
- transformers/modeling_utils.py +611 -527
- transformers/models/__init__.py +22 -0
- transformers/models/afmoe/modeling_afmoe.py +10 -19
- transformers/models/afmoe/modular_afmoe.py +5 -13
- transformers/models/aimv2/modeling_aimv2.py +4 -0
- transformers/models/aimv2/modular_aimv2.py +4 -0
- transformers/models/albert/modeling_albert.py +3 -0
- transformers/models/albert/tokenization_albert.py +6 -12
- transformers/models/align/modeling_align.py +14 -6
- transformers/models/altclip/modeling_altclip.py +11 -3
- transformers/models/apertus/modeling_apertus.py +8 -6
- transformers/models/apertus/modular_apertus.py +4 -1
- transformers/models/arcee/modeling_arcee.py +5 -5
- transformers/models/aria/modeling_aria.py +12 -8
- transformers/models/aria/modular_aria.py +7 -3
- transformers/models/audioflamingo3/modeling_audioflamingo3.py +1 -0
- transformers/models/audioflamingo3/modular_audioflamingo3.py +1 -0
- transformers/models/audioflamingo3/processing_audioflamingo3.py +27 -22
- transformers/models/auto/auto_factory.py +1 -1
- transformers/models/auto/configuration_auto.py +38 -0
- transformers/models/auto/feature_extraction_auto.py +9 -3
- transformers/models/auto/image_processing_auto.py +5 -2
- transformers/models/auto/modeling_auto.py +37 -0
- transformers/models/auto/processing_auto.py +22 -10
- transformers/models/auto/tokenization_auto.py +147 -566
- transformers/models/auto/video_processing_auto.py +5 -2
- transformers/models/autoformer/modeling_autoformer.py +4 -0
- transformers/models/aya_vision/modeling_aya_vision.py +7 -3
- transformers/models/bamba/modeling_bamba.py +21 -21
- transformers/models/bamba/modular_bamba.py +17 -16
- transformers/models/bark/modeling_bark.py +11 -0
- transformers/models/bart/configuration_bart.py +0 -1
- transformers/models/bart/modeling_bart.py +14 -0
- transformers/models/barthez/tokenization_barthez.py +5 -10
- transformers/models/beit/image_processing_beit_fast.py +0 -1
- transformers/models/beit/modeling_beit.py +6 -1
- transformers/models/bert/modeling_bert.py +3 -0
- transformers/models/bert/tokenization_bert.py +8 -21
- transformers/models/bert_generation/modeling_bert_generation.py +2 -0
- transformers/models/big_bird/modeling_big_bird.py +9 -0
- transformers/models/big_bird/tokenization_big_bird.py +18 -42
- transformers/models/bigbird_pegasus/modeling_bigbird_pegasus.py +15 -2
- transformers/models/biogpt/modeling_biogpt.py +2 -0
- transformers/models/biogpt/modular_biogpt.py +2 -0
- transformers/models/bit/modeling_bit.py +16 -3
- transformers/models/bitnet/modeling_bitnet.py +5 -5
- transformers/models/blenderbot/modeling_blenderbot.py +12 -0
- transformers/models/blenderbot/tokenization_blenderbot.py +18 -23
- transformers/models/blenderbot_small/modeling_blenderbot_small.py +12 -0
- transformers/models/blip/modeling_blip.py +2 -0
- transformers/models/blip/modeling_blip_text.py +10 -0
- transformers/models/blip_2/modeling_blip_2.py +4 -1
- transformers/models/bloom/modeling_bloom.py +17 -44
- transformers/models/blt/modeling_blt.py +164 -4
- transformers/models/blt/modular_blt.py +170 -5
- transformers/models/bridgetower/image_processing_bridgetower_fast.py +0 -2
- transformers/models/bridgetower/modeling_bridgetower.py +11 -1
- transformers/models/bros/modeling_bros.py +12 -0
- transformers/models/camembert/modeling_camembert.py +109 -106
- transformers/models/camembert/tokenization_camembert.py +8 -12
- transformers/models/canine/modeling_canine.py +11 -0
- transformers/models/canine/tokenization_canine.py +2 -0
- transformers/models/chameleon/modeling_chameleon.py +11 -5
- transformers/models/chinese_clip/modeling_chinese_clip.py +9 -3
- transformers/models/clap/feature_extraction_clap.py +2 -2
- transformers/models/clap/modeling_clap.py +30 -15
- transformers/models/clip/modeling_clip.py +2 -0
- transformers/models/clip/tokenization_clip.py +22 -44
- transformers/models/clipseg/modeling_clipseg.py +9 -0
- transformers/models/clvp/modeling_clvp.py +19 -3
- transformers/models/clvp/tokenization_clvp.py +1 -63
- transformers/models/code_llama/tokenization_code_llama.py +20 -43
- transformers/models/codegen/modeling_codegen.py +13 -4
- transformers/models/codegen/tokenization_codegen.py +14 -43
- transformers/models/cohere/modeling_cohere.py +5 -4
- transformers/models/cohere/modular_cohere.py +2 -1
- transformers/models/cohere/tokenization_cohere.py +12 -42
- transformers/models/cohere2/modeling_cohere2.py +8 -7
- transformers/models/cohere2/modular_cohere2.py +5 -5
- transformers/models/cohere2_vision/image_processing_cohere2_vision_fast.py +4 -4
- transformers/models/cohere2_vision/modeling_cohere2_vision.py +7 -3
- transformers/models/cohere2_vision/modular_cohere2_vision.py +4 -3
- transformers/models/colqwen2/modeling_colqwen2.py +1 -0
- transformers/models/colqwen2/modular_colqwen2.py +1 -0
- transformers/models/conditional_detr/configuration_conditional_detr.py +1 -1
- transformers/models/conditional_detr/modeling_conditional_detr.py +9 -1
- transformers/models/convbert/modeling_convbert.py +9 -0
- transformers/models/convnext/image_processing_convnext.py +2 -2
- transformers/models/convnext/image_processing_convnext_fast.py +9 -13
- transformers/models/convnext/modeling_convnext.py +2 -4
- transformers/models/convnextv2/modeling_convnextv2.py +2 -4
- transformers/models/csm/generation_csm.py +19 -22
- transformers/models/csm/modeling_csm.py +7 -4
- transformers/models/csm/modular_csm.py +2 -0
- transformers/models/ctrl/modeling_ctrl.py +15 -2
- transformers/models/cvt/modeling_cvt.py +7 -1
- transformers/models/cwm/modeling_cwm.py +5 -5
- transformers/models/d_fine/configuration_d_fine.py +3 -4
- transformers/models/d_fine/modeling_d_fine.py +48 -39
- transformers/models/d_fine/modular_d_fine.py +16 -4
- transformers/models/dab_detr/configuration_dab_detr.py +2 -2
- transformers/models/dab_detr/modeling_dab_detr.py +5 -1
- transformers/models/dac/modeling_dac.py +6 -6
- transformers/models/data2vec/modeling_data2vec_audio.py +5 -0
- transformers/models/data2vec/modeling_data2vec_text.py +7 -0
- transformers/models/data2vec/modeling_data2vec_vision.py +4 -1
- transformers/models/data2vec/modular_data2vec_text.py +7 -0
- transformers/models/dbrx/configuration_dbrx.py +9 -1
- transformers/models/dbrx/modeling_dbrx.py +3 -3
- transformers/models/deberta/modeling_deberta.py +7 -0
- transformers/models/deberta/tokenization_deberta.py +11 -20
- transformers/models/deberta_v2/modeling_deberta_v2.py +8 -0
- transformers/models/deberta_v2/tokenization_deberta_v2.py +13 -28
- transformers/models/decision_transformer/modeling_decision_transformer.py +12 -6
- transformers/models/deepseek_v2/modeling_deepseek_v2.py +9 -7
- transformers/models/deepseek_v2/modular_deepseek_v2.py +6 -4
- transformers/models/deepseek_v3/modeling_deepseek_v3.py +12 -7
- transformers/models/deepseek_v3/modular_deepseek_v3.py +7 -2
- transformers/models/deepseek_vl/image_processing_deepseek_vl_fast.py +0 -1
- transformers/models/deepseek_vl/modeling_deepseek_vl.py +9 -5
- transformers/models/deepseek_vl/modular_deepseek_vl.py +3 -0
- transformers/models/deepseek_vl_hybrid/image_processing_deepseek_vl_hybrid_fast.py +0 -4
- transformers/models/deepseek_vl_hybrid/modeling_deepseek_vl_hybrid.py +9 -5
- transformers/models/deepseek_vl_hybrid/modular_deepseek_vl_hybrid.py +9 -9
- transformers/models/deformable_detr/configuration_deformable_detr.py +2 -2
- transformers/models/deformable_detr/modeling_deformable_detr.py +5 -1
- transformers/models/depth_anything/configuration_depth_anything.py +2 -3
- transformers/models/depth_anything/modeling_depth_anything.py +1 -0
- transformers/models/depth_pro/image_processing_depth_pro_fast.py +0 -1
- transformers/models/depth_pro/modeling_depth_pro.py +2 -0
- transformers/models/detr/configuration_detr.py +1 -1
- transformers/models/detr/modeling_detr.py +13 -1
- transformers/models/dia/generation_dia.py +3 -10
- transformers/models/dia/modeling_dia.py +16 -4
- transformers/models/dia/modular_dia.py +11 -1
- transformers/models/dia/processing_dia.py +1 -1
- transformers/models/diffllama/modeling_diffllama.py +5 -5
- transformers/models/diffllama/modular_diffllama.py +2 -2
- transformers/models/dinat/modeling_dinat.py +3 -0
- transformers/models/dinov3_convnext/modeling_dinov3_convnext.py +1 -1
- transformers/models/dinov3_vit/image_processing_dinov3_vit_fast.py +0 -1
- transformers/models/dinov3_vit/modeling_dinov3_vit.py +5 -2
- transformers/models/dinov3_vit/modular_dinov3_vit.py +5 -2
- transformers/models/distilbert/modeling_distilbert.py +11 -9
- transformers/models/distilbert/tokenization_distilbert.py +13 -0
- transformers/models/doge/modeling_doge.py +3 -4
- transformers/models/doge/modular_doge.py +0 -1
- transformers/models/donut/image_processing_donut_fast.py +0 -1
- transformers/models/donut/modeling_donut_swin.py +18 -12
- transformers/models/dots1/modeling_dots1.py +23 -11
- transformers/models/dots1/modular_dots1.py +5 -3
- transformers/models/dpr/modeling_dpr.py +5 -0
- transformers/models/dpr/tokenization_dpr.py +12 -0
- transformers/models/dpt/configuration_dpt.py +1 -1
- transformers/models/dpt/image_processing_dpt_fast.py +1 -2
- transformers/models/dpt/modular_dpt.py +1 -2
- transformers/models/edgetam/configuration_edgetam.py +1 -1
- transformers/models/edgetam/modeling_edgetam.py +6 -3
- transformers/models/edgetam/modular_edgetam.py +15 -14
- transformers/models/edgetam_video/modeling_edgetam_video.py +56 -43
- transformers/models/edgetam_video/modular_edgetam_video.py +14 -19
- transformers/models/efficientloftr/image_processing_efficientloftr_fast.py +1 -2
- transformers/models/efficientloftr/modeling_efficientloftr.py +16 -3
- transformers/models/efficientnet/image_processing_efficientnet.py +5 -6
- transformers/models/efficientnet/image_processing_efficientnet_fast.py +1 -2
- transformers/models/efficientnet/modeling_efficientnet.py +7 -1
- transformers/models/electra/modeling_electra.py +7 -0
- transformers/models/emu3/modeling_emu3.py +12 -6
- transformers/models/emu3/modular_emu3.py +7 -1
- transformers/models/encodec/modeling_encodec.py +14 -0
- transformers/models/eomt/image_processing_eomt.py +13 -1
- transformers/models/eomt/image_processing_eomt_fast.py +60 -16
- transformers/models/eomt/modeling_eomt.py +7 -0
- transformers/models/eomt/modular_eomt.py +7 -0
- transformers/models/ernie/modeling_ernie.py +6 -0
- transformers/models/ernie/modular_ernie.py +6 -0
- transformers/models/ernie4_5/modeling_ernie4_5.py +5 -5
- transformers/models/ernie4_5/modular_ernie4_5.py +2 -1
- transformers/models/ernie4_5_moe/modeling_ernie4_5_moe.py +20 -17
- transformers/models/ernie4_5_moe/modular_ernie4_5_moe.py +11 -37
- transformers/models/ernie4_5_vl_moe/__init__.py +31 -0
- transformers/models/ernie4_5_vl_moe/configuration_ernie4_5_vl_moe.py +330 -0
- transformers/models/ernie4_5_vl_moe/image_processing_ernie4_5_vl_moe.py +456 -0
- transformers/models/ernie4_5_vl_moe/image_processing_ernie4_5_vl_moe_fast.py +232 -0
- transformers/models/ernie4_5_vl_moe/modeling_ernie4_5_vl_moe.py +1898 -0
- transformers/models/ernie4_5_vl_moe/modular_ernie4_5_vl_moe.py +1904 -0
- transformers/models/ernie4_5_vl_moe/processing_ernie4_5_vl_moe.py +251 -0
- transformers/models/ernie4_5_vl_moe/video_processing_ernie4_5_vl_moe.py +594 -0
- transformers/models/esm/modeling_esm.py +6 -0
- transformers/models/esm/modeling_esmfold.py +11 -5
- transformers/models/evolla/modeling_evolla.py +13 -5
- transformers/models/evolla/modular_evolla.py +8 -0
- transformers/models/exaone4/modeling_exaone4.py +3 -3
- transformers/models/exaone4/modular_exaone4.py +0 -1
- transformers/models/falcon/modeling_falcon.py +9 -4
- transformers/models/falcon_h1/modeling_falcon_h1.py +32 -26
- transformers/models/falcon_h1/modular_falcon_h1.py +7 -2
- transformers/models/falcon_mamba/modeling_falcon_mamba.py +31 -37
- transformers/models/falcon_mamba/modular_falcon_mamba.py +19 -33
- transformers/models/fast_vlm/__init__.py +27 -0
- transformers/models/fast_vlm/configuration_fast_vlm.py +137 -0
- transformers/models/fast_vlm/modeling_fast_vlm.py +459 -0
- transformers/models/fast_vlm/modular_fast_vlm.py +273 -0
- transformers/models/fastspeech2_conformer/modeling_fastspeech2_conformer.py +31 -13
- transformers/models/fastspeech2_conformer/tokenization_fastspeech2_conformer.py +1 -0
- transformers/models/flaubert/modeling_flaubert.py +21 -15
- transformers/models/flava/image_processing_flava_fast.py +0 -2
- transformers/models/flava/modeling_flava.py +10 -2
- transformers/models/flex_olmo/modeling_flex_olmo.py +10 -8
- transformers/models/florence2/modeling_florence2.py +22 -4
- transformers/models/florence2/modular_florence2.py +15 -1
- transformers/models/fnet/modeling_fnet.py +14 -0
- transformers/models/focalnet/modeling_focalnet.py +4 -0
- transformers/models/fsmt/modeling_fsmt.py +2 -0
- transformers/models/funnel/modeling_funnel.py +8 -0
- transformers/models/funnel/tokenization_funnel.py +17 -24
- transformers/models/fuyu/image_processing_fuyu.py +1 -1
- transformers/models/fuyu/modeling_fuyu.py +3 -1
- transformers/models/fuyu/processing_fuyu.py +19 -3
- transformers/models/gemma/modeling_gemma.py +14 -16
- transformers/models/gemma/modular_gemma.py +9 -11
- transformers/models/gemma/tokenization_gemma.py +10 -27
- transformers/models/gemma2/modeling_gemma2.py +5 -5
- transformers/models/gemma2/modular_gemma2.py +3 -2
- transformers/models/gemma3/image_processing_gemma3_fast.py +0 -1
- transformers/models/gemma3/modeling_gemma3.py +42 -91
- transformers/models/gemma3/modular_gemma3.py +38 -87
- transformers/models/gemma3n/configuration_gemma3n.py +3 -0
- transformers/models/gemma3n/modeling_gemma3n.py +65 -218
- transformers/models/gemma3n/modular_gemma3n.py +68 -68
- transformers/models/git/modeling_git.py +183 -126
- transformers/models/glm/modeling_glm.py +5 -5
- transformers/models/glm4/modeling_glm4.py +5 -5
- transformers/models/glm46v/image_processing_glm46v.py +0 -4
- transformers/models/glm46v/modeling_glm46v.py +3 -1
- transformers/models/glm46v/modular_glm46v.py +3 -0
- transformers/models/glm4_moe/modeling_glm4_moe.py +13 -7
- transformers/models/glm4_moe/modular_glm4_moe.py +1 -1
- transformers/models/glm4v/configuration_glm4v.py +3 -1
- transformers/models/glm4v/image_processing_glm4v.py +0 -4
- transformers/models/glm4v/modeling_glm4v.py +18 -8
- transformers/models/glm4v/modular_glm4v.py +17 -7
- transformers/models/glm4v_moe/configuration_glm4v_moe.py +3 -1
- transformers/models/glm4v_moe/modeling_glm4v_moe.py +44 -27
- transformers/models/glm4v_moe/modular_glm4v_moe.py +13 -1
- transformers/models/glmasr/__init__.py +30 -0
- transformers/models/glmasr/configuration_glmasr.py +197 -0
- transformers/models/glmasr/modeling_glmasr.py +512 -0
- transformers/models/glmasr/modular_glmasr.py +433 -0
- transformers/models/glmasr/processing_glmasr.py +332 -0
- transformers/models/glpn/image_processing_glpn_fast.py +0 -1
- transformers/models/glpn/modeling_glpn.py +2 -0
- transformers/models/got_ocr2/image_processing_got_ocr2_fast.py +0 -1
- transformers/models/got_ocr2/modeling_got_ocr2.py +8 -3
- transformers/models/gpt2/modeling_gpt2.py +13 -6
- transformers/models/gpt2/tokenization_gpt2.py +16 -44
- transformers/models/gpt_bigcode/modeling_gpt_bigcode.py +4 -8
- transformers/models/gpt_neo/modeling_gpt_neo.py +19 -3
- transformers/models/gpt_neox/modeling_gpt_neox.py +6 -3
- transformers/models/gpt_neox/modular_gpt_neox.py +3 -0
- transformers/models/gpt_neox/tokenization_gpt_neox.py +10 -49
- transformers/models/gpt_neox_japanese/modeling_gpt_neox_japanese.py +4 -2
- transformers/models/gpt_oss/configuration_gpt_oss.py +17 -0
- transformers/models/gpt_oss/modeling_gpt_oss.py +10 -14
- transformers/models/gpt_oss/modular_gpt_oss.py +8 -12
- transformers/models/gptj/modeling_gptj.py +18 -6
- transformers/models/granite/modeling_granite.py +5 -5
- transformers/models/granite_speech/modeling_granite_speech.py +15 -1
- transformers/models/granitemoe/modeling_granitemoe.py +6 -9
- transformers/models/granitemoe/modular_granitemoe.py +1 -4
- transformers/models/granitemoehybrid/configuration_granitemoehybrid.py +4 -0
- transformers/models/granitemoehybrid/modeling_granitemoehybrid.py +36 -28
- transformers/models/granitemoehybrid/modular_granitemoehybrid.py +12 -2
- transformers/models/granitemoeshared/modeling_granitemoeshared.py +6 -9
- transformers/models/grounding_dino/configuration_grounding_dino.py +2 -3
- transformers/models/grounding_dino/modeling_grounding_dino.py +8 -4
- transformers/models/groupvit/modeling_groupvit.py +9 -1
- transformers/models/helium/modeling_helium.py +5 -4
- transformers/models/herbert/tokenization_herbert.py +9 -25
- transformers/models/hgnet_v2/modeling_hgnet_v2.py +16 -1
- transformers/models/hgnet_v2/modular_hgnet_v2.py +16 -1
- transformers/models/hiera/modeling_hiera.py +4 -0
- transformers/models/hubert/modeling_hubert.py +7 -0
- transformers/models/hubert/modular_hubert.py +5 -0
- transformers/models/hunyuan_v1_dense/modeling_hunyuan_v1_dense.py +5 -5
- transformers/models/hunyuan_v1_dense/modular_hunyuan_v1_dense.py +1 -1
- transformers/models/hunyuan_v1_moe/__init__.py +1 -1
- transformers/models/hunyuan_v1_moe/modeling_hunyuan_v1_moe.py +15 -7
- transformers/models/hunyuan_v1_moe/modular_hunyuan_v1_moe.py +4 -2
- transformers/models/ibert/modeling_ibert.py +22 -0
- transformers/models/idefics/modeling_idefics.py +15 -21
- transformers/models/idefics2/modeling_idefics2.py +7 -1
- transformers/models/idefics3/modeling_idefics3.py +5 -1
- transformers/models/imagegpt/image_processing_imagegpt_fast.py +1 -5
- transformers/models/imagegpt/modeling_imagegpt.py +11 -3
- transformers/models/informer/modeling_informer.py +4 -0
- transformers/models/informer/modular_informer.py +1 -0
- transformers/models/instructblip/modeling_instructblip.py +2 -0
- transformers/models/instructblipvideo/modeling_instructblipvideo.py +52 -50
- transformers/models/instructblipvideo/video_processing_instructblipvideo.py +0 -1
- transformers/models/internvl/modeling_internvl.py +13 -12
- transformers/models/internvl/modular_internvl.py +7 -13
- transformers/models/internvl/video_processing_internvl.py +0 -1
- transformers/models/jais2/__init__.py +27 -0
- transformers/models/jais2/configuration_jais2.py +152 -0
- transformers/models/jais2/modeling_jais2.py +486 -0
- transformers/models/jais2/modular_jais2.py +196 -0
- transformers/models/jamba/modeling_jamba.py +25 -20
- transformers/models/jamba/modular_jamba.py +17 -17
- transformers/models/janus/image_processing_janus_fast.py +0 -1
- transformers/models/janus/modeling_janus.py +16 -7
- transformers/models/janus/modular_janus.py +17 -7
- transformers/models/jetmoe/modeling_jetmoe.py +4 -4
- transformers/models/jetmoe/modular_jetmoe.py +1 -0
- transformers/models/kosmos2/modeling_kosmos2.py +15 -2
- transformers/models/kosmos2_5/image_processing_kosmos2_5_fast.py +2 -2
- transformers/models/kosmos2_5/modeling_kosmos2_5.py +10 -1
- transformers/models/kyutai_speech_to_text/modeling_kyutai_speech_to_text.py +12 -4
- transformers/models/kyutai_speech_to_text/modular_kyutai_speech_to_text.py +9 -1
- transformers/models/lasr/__init__.py +29 -0
- transformers/models/lasr/configuration_lasr.py +248 -0
- transformers/models/lasr/feature_extraction_lasr.py +277 -0
- transformers/models/lasr/modeling_lasr.py +730 -0
- transformers/models/lasr/modular_lasr.py +576 -0
- transformers/models/lasr/processing_lasr.py +94 -0
- transformers/models/lasr/tokenization_lasr.py +186 -0
- transformers/models/layoutlm/modeling_layoutlm.py +10 -3
- transformers/models/layoutlmv2/image_processing_layoutlmv2_fast.py +0 -1
- transformers/models/layoutlmv2/modeling_layoutlmv2.py +16 -0
- transformers/models/layoutlmv2/tokenization_layoutlmv2.py +11 -53
- transformers/models/layoutlmv3/image_processing_layoutlmv3_fast.py +0 -1
- transformers/models/layoutlmv3/modeling_layoutlmv3.py +33 -5
- transformers/models/layoutlmv3/tokenization_layoutlmv3.py +12 -61
- transformers/models/layoutxlm/tokenization_layoutxlm.py +13 -38
- transformers/models/led/modeling_led.py +12 -0
- transformers/models/levit/modeling_levit.py +21 -0
- transformers/models/lfm2/modeling_lfm2.py +5 -6
- transformers/models/lfm2/modular_lfm2.py +0 -1
- transformers/models/lfm2_moe/modeling_lfm2_moe.py +17 -8
- transformers/models/lfm2_moe/modular_lfm2_moe.py +5 -28
- transformers/models/lfm2_vl/configuration_lfm2_vl.py +4 -0
- transformers/models/lfm2_vl/modeling_lfm2_vl.py +11 -5
- transformers/models/lfm2_vl/modular_lfm2_vl.py +4 -2
- transformers/models/lfm2_vl/processing_lfm2_vl.py +82 -42
- transformers/models/lightglue/image_processing_lightglue_fast.py +1 -2
- transformers/models/lightglue/modeling_lightglue.py +3 -1
- transformers/models/lightglue/modular_lightglue.py +1 -0
- transformers/models/lilt/modeling_lilt.py +23 -15
- transformers/models/llama/modeling_llama.py +5 -5
- transformers/models/llama/tokenization_llama.py +15 -43
- transformers/models/llama4/image_processing_llama4_fast.py +1 -2
- transformers/models/llama4/modeling_llama4.py +11 -6
- transformers/models/llava/image_processing_llava_fast.py +0 -1
- transformers/models/llava/modeling_llava.py +12 -7
- transformers/models/llava_next/image_processing_llava_next_fast.py +0 -1
- transformers/models/llava_next/modeling_llava_next.py +7 -3
- transformers/models/llava_next_video/modeling_llava_next_video.py +7 -3
- transformers/models/llava_next_video/modular_llava_next_video.py +7 -3
- transformers/models/llava_onevision/image_processing_llava_onevision_fast.py +0 -1
- transformers/models/llava_onevision/modeling_llava_onevision.py +7 -3
- transformers/models/llava_onevision/modular_llava_onevision.py +7 -4
- transformers/models/longcat_flash/modeling_longcat_flash.py +6 -5
- transformers/models/longcat_flash/modular_longcat_flash.py +3 -2
- transformers/models/longformer/modeling_longformer.py +6 -0
- transformers/models/longt5/modeling_longt5.py +4 -4
- transformers/models/luke/modeling_luke.py +9 -0
- transformers/models/luke/tokenization_luke.py +11 -38
- transformers/models/lxmert/modeling_lxmert.py +2 -0
- transformers/models/m2m_100/modeling_m2m_100.py +14 -0
- transformers/models/mamba/modeling_mamba.py +16 -23
- transformers/models/mamba2/modeling_mamba2.py +24 -23
- transformers/models/marian/configuration_marian.py +1 -1
- transformers/models/marian/modeling_marian.py +8 -0
- transformers/models/markuplm/modeling_markuplm.py +9 -8
- transformers/models/markuplm/tokenization_markuplm.py +28 -61
- transformers/models/mask2former/configuration_mask2former.py +3 -3
- transformers/models/mask2former/image_processing_mask2former_fast.py +1 -4
- transformers/models/mask2former/modeling_mask2former.py +11 -0
- transformers/models/maskformer/configuration_maskformer.py +3 -3
- transformers/models/maskformer/image_processing_maskformer_fast.py +1 -4
- transformers/models/maskformer/modeling_maskformer.py +11 -1
- transformers/models/maskformer/modeling_maskformer_swin.py +21 -15
- transformers/models/mbart/configuration_mbart.py +1 -0
- transformers/models/mbart/modeling_mbart.py +14 -0
- transformers/models/mbart/tokenization_mbart.py +11 -52
- transformers/models/mbart50/tokenization_mbart50.py +7 -10
- transformers/models/megatron_bert/modeling_megatron_bert.py +9 -0
- transformers/models/metaclip_2/modeling_metaclip_2.py +2 -0
- transformers/models/metaclip_2/modular_metaclip_2.py +2 -0
- transformers/models/mgp_str/modeling_mgp_str.py +2 -0
- transformers/models/mimi/modeling_mimi.py +28 -5
- transformers/models/minimax/modeling_minimax.py +19 -6
- transformers/models/minimax/modular_minimax.py +12 -1
- transformers/models/ministral/modeling_ministral.py +5 -5
- transformers/models/ministral3/configuration_ministral3.py +1 -1
- transformers/models/ministral3/modeling_ministral3.py +5 -4
- transformers/models/mistral/modeling_mistral.py +5 -4
- transformers/models/mistral3/modeling_mistral3.py +10 -4
- transformers/models/mistral3/modular_mistral3.py +3 -1
- transformers/models/mixtral/modeling_mixtral.py +15 -7
- transformers/models/mixtral/modular_mixtral.py +6 -2
- transformers/models/mlcd/modeling_mlcd.py +6 -0
- transformers/models/mlcd/modular_mlcd.py +4 -0
- transformers/models/mllama/modeling_mllama.py +15 -4
- transformers/models/mluke/tokenization_mluke.py +6 -6
- transformers/models/mm_grounding_dino/configuration_mm_grounding_dino.py +1 -2
- transformers/models/mm_grounding_dino/modeling_mm_grounding_dino.py +8 -4
- transformers/models/mm_grounding_dino/modular_mm_grounding_dino.py +1 -2
- transformers/models/mobilebert/modeling_mobilebert.py +2 -0
- transformers/models/mobilenet_v1/modeling_mobilenet_v1.py +2 -0
- transformers/models/mobilenet_v2/image_processing_mobilenet_v2_fast.py +0 -1
- transformers/models/mobilenet_v2/modeling_mobilenet_v2.py +3 -0
- transformers/models/mobilevit/image_processing_mobilevit.py +5 -5
- transformers/models/mobilevit/image_processing_mobilevit_fast.py +1 -2
- transformers/models/mobilevit/modeling_mobilevit.py +7 -0
- transformers/models/mobilevitv2/modeling_mobilevitv2.py +7 -0
- transformers/models/modernbert/modeling_modernbert.py +16 -2
- transformers/models/modernbert/modular_modernbert.py +14 -1
- transformers/models/modernbert_decoder/modeling_modernbert_decoder.py +17 -10
- transformers/models/modernbert_decoder/modular_modernbert_decoder.py +15 -8
- transformers/models/moonshine/modeling_moonshine.py +5 -3
- transformers/models/moshi/modeling_moshi.py +26 -53
- transformers/models/mpnet/modeling_mpnet.py +7 -0
- transformers/models/mpnet/tokenization_mpnet.py +5 -13
- transformers/models/mpt/modeling_mpt.py +2 -0
- transformers/models/mra/modeling_mra.py +10 -1
- transformers/models/mt5/configuration_mt5.py +2 -3
- transformers/models/mt5/modeling_mt5.py +7 -10
- transformers/models/musicgen/modeling_musicgen.py +7 -9
- transformers/models/musicgen_melody/modeling_musicgen_melody.py +7 -0
- transformers/models/mvp/modeling_mvp.py +14 -0
- transformers/models/nanochat/modeling_nanochat.py +5 -5
- transformers/models/nemotron/modeling_nemotron.py +7 -5
- transformers/models/nllb/tokenization_nllb.py +8 -22
- transformers/models/nllb_moe/configuration_nllb_moe.py +1 -0
- transformers/models/nllb_moe/modeling_nllb_moe.py +10 -0
- transformers/models/nougat/image_processing_nougat_fast.py +0 -1
- transformers/models/nougat/tokenization_nougat.py +15 -68
- transformers/models/nystromformer/modeling_nystromformer.py +13 -0
- transformers/models/olmo/modeling_olmo.py +5 -5
- transformers/models/olmo/modular_olmo.py +2 -2
- transformers/models/olmo2/modeling_olmo2.py +5 -6
- transformers/models/olmo2/modular_olmo2.py +0 -1
- transformers/models/olmo3/modeling_olmo3.py +5 -5
- transformers/models/olmoe/modeling_olmoe.py +15 -7
- transformers/models/olmoe/modular_olmoe.py +4 -2
- transformers/models/omdet_turbo/configuration_omdet_turbo.py +2 -2
- transformers/models/omdet_turbo/modeling_omdet_turbo.py +6 -0
- transformers/models/oneformer/configuration_oneformer.py +3 -3
- transformers/models/oneformer/modeling_oneformer.py +11 -39
- transformers/models/openai/modeling_openai.py +15 -0
- transformers/models/openai/tokenization_openai.py +10 -46
- transformers/models/opt/modeling_opt.py +2 -0
- transformers/models/ovis2/image_processing_ovis2_fast.py +0 -1
- transformers/models/ovis2/modeling_ovis2.py +15 -3
- transformers/models/ovis2/modular_ovis2.py +8 -0
- transformers/models/owlv2/image_processing_owlv2_fast.py +0 -2
- transformers/models/owlv2/modeling_owlv2.py +11 -3
- transformers/models/owlv2/modular_owlv2.py +0 -2
- transformers/models/owlvit/modeling_owlvit.py +11 -3
- transformers/models/paddleocr_vl/__init__.py +32 -0
- transformers/models/paddleocr_vl/configuration_paddleocr_vl.py +336 -0
- transformers/models/paddleocr_vl/image_processing_paddleocr_vl.py +504 -0
- transformers/models/paddleocr_vl/image_processing_paddleocr_vl_fast.py +209 -0
- transformers/models/paddleocr_vl/modeling_paddleocr_vl.py +1682 -0
- transformers/models/paddleocr_vl/modular_paddleocr_vl.py +1359 -0
- transformers/models/paddleocr_vl/processing_paddleocr_vl.py +135 -0
- transformers/models/paligemma/modeling_paligemma.py +25 -17
- transformers/models/parakeet/configuration_parakeet.py +4 -6
- transformers/models/parakeet/modeling_parakeet.py +14 -6
- transformers/models/parakeet/modular_parakeet.py +7 -2
- transformers/models/parakeet/processing_parakeet.py +1 -0
- transformers/models/parakeet/{tokenization_parakeet_fast.py → tokenization_parakeet.py} +3 -3
- transformers/models/patchtsmixer/modeling_patchtsmixer.py +10 -0
- transformers/models/patchtst/modeling_patchtst.py +25 -6
- transformers/models/pe_audio/__init__.py +30 -0
- transformers/models/pe_audio/configuration_pe_audio.py +206 -0
- transformers/models/pe_audio/feature_extraction_pe_audio.py +162 -0
- transformers/models/pe_audio/modeling_pe_audio.py +820 -0
- transformers/models/pe_audio/modular_pe_audio.py +299 -0
- transformers/{kernels/falcon_mamba/__init__.py → models/pe_audio/processing_pe_audio.py} +11 -2
- transformers/models/pe_audio_video/__init__.py +29 -0
- transformers/models/pe_audio_video/configuration_pe_audio_video.py +225 -0
- transformers/models/pe_audio_video/modeling_pe_audio_video.py +972 -0
- transformers/models/pe_audio_video/modular_pe_audio_video.py +764 -0
- transformers/models/pe_audio_video/processing_pe_audio_video.py +25 -0
- transformers/models/pe_video/__init__.py +30 -0
- transformers/models/pe_video/configuration_pe_video.py +211 -0
- transformers/models/pe_video/modeling_pe_video.py +636 -0
- transformers/models/pe_video/modular_pe_video.py +219 -0
- transformers/models/pe_video/processing_pe_video.py +10 -0
- transformers/models/pe_video/video_processing_pe_video.py +66 -0
- transformers/models/pegasus/configuration_pegasus.py +1 -0
- transformers/models/pegasus/modeling_pegasus.py +8 -0
- transformers/models/pegasus/tokenization_pegasus.py +17 -44
- transformers/models/pegasus_x/modeling_pegasus_x.py +5 -0
- transformers/models/perceiver/image_processing_perceiver_fast.py +0 -1
- transformers/models/perceiver/modeling_perceiver.py +13 -1
- transformers/models/perception_lm/image_processing_perception_lm_fast.py +0 -1
- transformers/models/perception_lm/modeling_perception_lm.py +7 -3
- transformers/models/perception_lm/modular_perception_lm.py +7 -3
- transformers/models/persimmon/modeling_persimmon.py +3 -2
- transformers/models/phi/modeling_phi.py +5 -6
- transformers/models/phi/modular_phi.py +0 -1
- transformers/models/phi3/modeling_phi3.py +3 -2
- transformers/models/phi4_multimodal/modeling_phi4_multimodal.py +9 -6
- transformers/models/phi4_multimodal/modular_phi4_multimodal.py +7 -4
- transformers/models/phi4_multimodal/processing_phi4_multimodal.py +0 -2
- transformers/models/phimoe/modeling_phimoe.py +15 -7
- transformers/models/phimoe/modular_phimoe.py +3 -3
- transformers/models/pix2struct/modeling_pix2struct.py +2 -0
- transformers/models/pix2struct/processing_pix2struct.py +0 -4
- transformers/models/pixio/__init__.py +30 -0
- transformers/models/pixio/configuration_pixio.py +151 -0
- transformers/models/pixio/modeling_pixio.py +507 -0
- transformers/models/pixio/modular_pixio.py +404 -0
- transformers/models/pixtral/modeling_pixtral.py +3 -2
- transformers/models/pixtral/processing_pixtral.py +3 -1
- transformers/models/plbart/configuration_plbart.py +1 -0
- transformers/models/plbart/modeling_plbart.py +13 -0
- transformers/models/plbart/modular_plbart.py +8 -0
- transformers/models/plbart/tokenization_plbart.py +0 -2
- transformers/models/poolformer/image_processing_poolformer_fast.py +0 -1
- transformers/models/poolformer/modeling_poolformer.py +13 -1
- transformers/models/pop2piano/configuration_pop2piano.py +0 -1
- transformers/models/pop2piano/modeling_pop2piano.py +2 -0
- transformers/models/prompt_depth_anything/configuration_prompt_depth_anything.py +2 -3
- transformers/models/prompt_depth_anything/modeling_prompt_depth_anything.py +1 -0
- transformers/models/prompt_depth_anything/modular_prompt_depth_anything.py +1 -0
- transformers/models/prophetnet/modeling_prophetnet.py +5 -1
- transformers/models/pvt/modeling_pvt.py +2 -0
- transformers/models/pvt_v2/modeling_pvt_v2.py +3 -0
- transformers/models/qwen2/modeling_qwen2.py +5 -5
- transformers/models/qwen2/tokenization_qwen2.py +14 -18
- transformers/models/qwen2_5_omni/configuration_qwen2_5_omni.py +4 -2
- transformers/models/qwen2_5_omni/modeling_qwen2_5_omni.py +116 -79
- transformers/models/qwen2_5_omni/modular_qwen2_5_omni.py +71 -33
- transformers/models/qwen2_5_vl/configuration_qwen2_5_vl.py +1 -1
- transformers/models/qwen2_5_vl/modeling_qwen2_5_vl.py +23 -11
- transformers/models/qwen2_5_vl/modular_qwen2_5_vl.py +29 -27
- transformers/models/qwen2_audio/modeling_qwen2_audio.py +4 -2
- transformers/models/qwen2_moe/modeling_qwen2_moe.py +15 -7
- transformers/models/qwen2_vl/configuration_qwen2_vl.py +1 -1
- transformers/models/qwen2_vl/image_processing_qwen2_vl.py +3 -2
- transformers/models/qwen2_vl/modeling_qwen2_vl.py +23 -20
- transformers/models/qwen3/modeling_qwen3.py +5 -5
- transformers/models/qwen3_moe/modeling_qwen3_moe.py +15 -7
- transformers/models/qwen3_next/modeling_qwen3_next.py +7 -8
- transformers/models/qwen3_omni_moe/configuration_qwen3_omni_moe.py +4 -0
- transformers/models/qwen3_omni_moe/modeling_qwen3_omni_moe.py +112 -68
- transformers/models/qwen3_omni_moe/modular_qwen3_omni_moe.py +62 -20
- transformers/models/qwen3_vl/configuration_qwen3_vl.py +5 -5
- transformers/models/qwen3_vl/modeling_qwen3_vl.py +57 -42
- transformers/models/qwen3_vl/modular_qwen3_vl.py +59 -46
- transformers/models/qwen3_vl/processing_qwen3_vl.py +3 -3
- transformers/models/qwen3_vl_moe/modeling_qwen3_vl_moe.py +132 -148
- transformers/models/qwen3_vl_moe/modular_qwen3_vl_moe.py +36 -82
- transformers/models/rag/configuration_rag.py +0 -8
- transformers/models/rag/modeling_rag.py +8 -9
- transformers/models/recurrent_gemma/modeling_recurrent_gemma.py +18 -3
- transformers/models/reformer/modeling_reformer.py +13 -1
- transformers/models/reformer/tokenization_reformer.py +11 -28
- transformers/models/regnet/modeling_regnet.py +10 -1
- transformers/models/rembert/modeling_rembert.py +13 -1
- transformers/models/rembert/tokenization_rembert.py +3 -10
- transformers/models/resnet/modeling_resnet.py +19 -5
- transformers/models/roberta/modeling_roberta.py +3 -0
- transformers/models/roberta/modular_roberta.py +3 -0
- transformers/models/roberta/tokenization_roberta.py +18 -27
- transformers/models/roberta_prelayernorm/modeling_roberta_prelayernorm.py +3 -0
- transformers/models/roc_bert/modeling_roc_bert.py +3 -0
- transformers/models/roformer/modeling_roformer.py +6 -0
- transformers/models/roformer/tokenization_roformer.py +77 -412
- transformers/models/rt_detr/configuration_rt_detr.py +1 -1
- transformers/models/rt_detr/modeling_rt_detr.py +6 -0
- transformers/models/rt_detr/modeling_rt_detr_resnet.py +13 -4
- transformers/models/rt_detr_v2/configuration_rt_detr_v2.py +2 -3
- transformers/models/rt_detr_v2/modeling_rt_detr_v2.py +9 -0
- transformers/models/rt_detr_v2/modular_rt_detr_v2.py +8 -3
- transformers/models/rwkv/modeling_rwkv.py +2 -1
- transformers/models/sam/configuration_sam.py +1 -0
- transformers/models/sam/image_processing_sam_fast.py +0 -1
- transformers/models/sam/modeling_sam.py +4 -1
- transformers/models/sam2/configuration_sam2.py +1 -1
- transformers/models/sam2/modeling_sam2.py +7 -3
- transformers/models/sam2/modular_sam2.py +7 -3
- transformers/models/sam2_video/modeling_sam2_video.py +52 -43
- transformers/models/sam2_video/modular_sam2_video.py +32 -18
- transformers/models/sam3/configuration_sam3.py +21 -1
- transformers/models/sam3/modeling_sam3.py +100 -80
- transformers/models/sam3_tracker/modeling_sam3_tracker.py +8 -1
- transformers/models/sam3_tracker/modular_sam3_tracker.py +8 -1
- transformers/models/sam3_tracker_video/configuration_sam3_tracker_video.py +25 -0
- transformers/models/sam3_tracker_video/modeling_sam3_tracker_video.py +27 -15
- transformers/models/sam3_tracker_video/modular_sam3_tracker_video.py +25 -2
- transformers/models/sam3_video/configuration_sam3_video.py +14 -0
- transformers/models/sam3_video/modeling_sam3_video.py +4 -3
- transformers/models/sam3_video/processing_sam3_video.py +1 -1
- transformers/models/sam_hq/configuration_sam_hq.py +1 -0
- transformers/models/sam_hq/modeling_sam_hq.py +26 -23
- transformers/models/seamless_m4t/modeling_seamless_m4t.py +32 -12
- transformers/models/seamless_m4t/tokenization_seamless_m4t.py +27 -59
- transformers/models/seamless_m4t_v2/modeling_seamless_m4t_v2.py +11 -1
- transformers/models/seed_oss/modeling_seed_oss.py +3 -3
- transformers/models/segformer/image_processing_segformer_fast.py +0 -1
- transformers/models/segformer/modeling_segformer.py +6 -3
- transformers/models/segformer/modular_segformer.py +0 -1
- transformers/models/seggpt/modeling_seggpt.py +2 -0
- transformers/models/sew/modeling_sew.py +3 -0
- transformers/models/sew/modular_sew.py +1 -0
- transformers/models/sew_d/modeling_sew_d.py +3 -0
- transformers/models/shieldgemma2/modeling_shieldgemma2.py +1 -0
- transformers/models/siglip/modeling_siglip.py +24 -2
- transformers/models/siglip2/modeling_siglip2.py +67 -41
- transformers/models/siglip2/modular_siglip2.py +4 -0
- transformers/models/smollm3/modeling_smollm3.py +5 -5
- transformers/models/smolvlm/modeling_smolvlm.py +5 -1
- transformers/models/smolvlm/processing_smolvlm.py +0 -7
- transformers/models/smolvlm/video_processing_smolvlm.py +0 -1
- transformers/models/speech_to_text/modeling_speech_to_text.py +14 -0
- transformers/models/speecht5/modeling_speecht5.py +41 -1
- transformers/models/splinter/modeling_splinter.py +12 -3
- transformers/models/splinter/tokenization_splinter.py +9 -28
- transformers/models/squeezebert/modeling_squeezebert.py +8 -0
- transformers/models/stablelm/modeling_stablelm.py +4 -2
- transformers/models/starcoder2/modeling_starcoder2.py +5 -4
- transformers/models/superglue/image_processing_superglue_fast.py +1 -2
- transformers/models/superglue/modeling_superglue.py +1 -0
- transformers/models/superpoint/image_processing_superpoint_fast.py +1 -2
- transformers/models/superpoint/modeling_superpoint.py +1 -0
- transformers/models/swiftformer/modeling_swiftformer.py +6 -0
- transformers/models/swin/modeling_swin.py +20 -12
- transformers/models/swin2sr/image_processing_swin2sr_fast.py +0 -1
- transformers/models/swin2sr/modeling_swin2sr.py +51 -33
- transformers/models/swinv2/modeling_swinv2.py +45 -33
- transformers/models/switch_transformers/modeling_switch_transformers.py +2 -8
- transformers/models/switch_transformers/modular_switch_transformers.py +2 -8
- transformers/models/t5/configuration_t5.py +7 -1
- transformers/models/t5/modeling_t5.py +8 -7
- transformers/models/t5/tokenization_t5.py +4 -8
- transformers/models/t5gemma/modeling_t5gemma.py +6 -6
- transformers/models/t5gemma2/configuration_t5gemma2.py +6 -42
- transformers/models/t5gemma2/modeling_t5gemma2.py +19 -10
- transformers/models/t5gemma2/modular_t5gemma2.py +289 -4
- transformers/models/table_transformer/configuration_table_transformer.py +1 -1
- transformers/models/table_transformer/modeling_table_transformer.py +5 -1
- transformers/models/tapas/modeling_tapas.py +3 -0
- transformers/models/textnet/image_processing_textnet_fast.py +0 -1
- transformers/models/textnet/modeling_textnet.py +11 -2
- transformers/models/time_series_transformer/modeling_time_series_transformer.py +4 -0
- transformers/models/timesfm/modeling_timesfm.py +14 -0
- transformers/models/timesfm/modular_timesfm.py +14 -0
- transformers/models/timesformer/modeling_timesformer.py +2 -0
- transformers/models/timm_backbone/modeling_timm_backbone.py +13 -9
- transformers/models/timm_wrapper/configuration_timm_wrapper.py +3 -0
- transformers/models/timm_wrapper/modeling_timm_wrapper.py +20 -14
- transformers/models/trocr/modeling_trocr.py +3 -2
- transformers/models/tvp/configuration_tvp.py +5 -1
- transformers/models/tvp/modeling_tvp.py +6 -4
- transformers/models/udop/configuration_udop.py +1 -0
- transformers/models/udop/modeling_udop.py +7 -7
- transformers/models/udop/tokenization_udop.py +5 -13
- transformers/models/umt5/configuration_umt5.py +2 -2
- transformers/models/umt5/modeling_umt5.py +7 -6
- transformers/models/unispeech/modeling_unispeech.py +4 -0
- transformers/models/unispeech/modular_unispeech.py +2 -0
- transformers/models/unispeech_sat/modeling_unispeech_sat.py +6 -0
- transformers/models/unispeech_sat/modular_unispeech_sat.py +2 -0
- transformers/models/univnet/modeling_univnet.py +1 -0
- transformers/models/upernet/modeling_upernet.py +1 -0
- transformers/models/vaultgemma/modeling_vaultgemma.py +5 -5
- transformers/models/video_llama_3/image_processing_video_llama_3.py +3 -2
- transformers/models/video_llama_3/modeling_video_llama_3.py +12 -1
- transformers/models/video_llama_3/modular_video_llama_3.py +10 -1
- transformers/models/video_llava/modeling_video_llava.py +7 -3
- transformers/models/vilt/configuration_vilt.py +2 -2
- transformers/models/vilt/modeling_vilt.py +13 -0
- transformers/models/vipllava/modeling_vipllava.py +7 -3
- transformers/models/vision_text_dual_encoder/modeling_vision_text_dual_encoder.py +1 -0
- transformers/models/visual_bert/modeling_visual_bert.py +8 -0
- transformers/models/vitdet/modeling_vitdet.py +2 -0
- transformers/models/vitmatte/configuration_vitmatte.py +1 -1
- transformers/models/vitmatte/image_processing_vitmatte_fast.py +0 -1
- transformers/models/vitmatte/modeling_vitmatte.py +5 -0
- transformers/models/vitpose/configuration_vitpose.py +1 -1
- transformers/models/vitpose/image_processing_vitpose_fast.py +0 -1
- transformers/models/vits/modeling_vits.py +1 -0
- transformers/models/vjepa2/modeling_vjepa2.py +1 -0
- transformers/models/voxtral/modeling_voxtral.py +2 -2
- transformers/models/voxtral/modular_voxtral.py +2 -2
- transformers/models/wav2vec2/modeling_wav2vec2.py +7 -0
- transformers/models/wav2vec2_bert/modeling_wav2vec2_bert.py +21 -10
- transformers/models/wav2vec2_bert/modular_wav2vec2_bert.py +12 -0
- transformers/models/wav2vec2_conformer/modeling_wav2vec2_conformer.py +27 -11
- transformers/models/wav2vec2_conformer/modular_wav2vec2_conformer.py +21 -11
- transformers/models/wavlm/modeling_wavlm.py +5 -0
- transformers/models/whisper/generation_whisper.py +1 -0
- transformers/models/whisper/modeling_whisper.py +11 -3
- transformers/models/whisper/tokenization_whisper.py +4 -15
- transformers/models/x_clip/modeling_x_clip.py +5 -0
- transformers/models/xcodec/modeling_xcodec.py +5 -0
- transformers/models/xglm/modeling_xglm.py +11 -0
- transformers/models/xglm/tokenization_xglm.py +4 -9
- transformers/models/xlm/modeling_xlm.py +18 -14
- transformers/models/xlm_roberta/modeling_xlm_roberta.py +109 -106
- transformers/models/xlm_roberta/tokenization_xlm_roberta.py +9 -16
- transformers/models/xlm_roberta_xl/modeling_xlm_roberta_xl.py +3 -0
- transformers/models/xlnet/modeling_xlnet.py +3 -1
- transformers/models/xlnet/tokenization_xlnet.py +3 -7
- transformers/models/xmod/modeling_xmod.py +3 -0
- transformers/models/yoso/modeling_yoso.py +10 -1
- transformers/models/zamba/modeling_zamba.py +4 -1
- transformers/models/zamba2/modeling_zamba2.py +7 -4
- transformers/models/zamba2/modular_zamba2.py +1 -1
- transformers/models/zoedepth/configuration_zoedepth.py +1 -1
- transformers/models/zoedepth/image_processing_zoedepth_fast.py +1 -3
- transformers/models/zoedepth/modeling_zoedepth.py +8 -0
- transformers/pipelines/__init__.py +11 -9
- transformers/pipelines/automatic_speech_recognition.py +20 -12
- transformers/pipelines/base.py +2 -10
- transformers/pipelines/document_question_answering.py +4 -2
- transformers/pipelines/question_answering.py +1 -1
- transformers/pipelines/text_generation.py +1 -1
- transformers/pipelines/text_to_audio.py +2 -2
- transformers/processing_utils.py +133 -50
- transformers/quantizers/auto.py +2 -4
- transformers/quantizers/base.py +44 -174
- transformers/quantizers/quantizer_aqlm.py +2 -23
- transformers/quantizers/quantizer_auto_round.py +2 -12
- transformers/quantizers/quantizer_awq.py +20 -89
- transformers/quantizers/quantizer_bitnet.py +4 -14
- transformers/quantizers/quantizer_bnb_4bit.py +18 -155
- transformers/quantizers/quantizer_bnb_8bit.py +24 -110
- transformers/quantizers/quantizer_compressed_tensors.py +2 -9
- transformers/quantizers/quantizer_eetq.py +16 -74
- transformers/quantizers/quantizer_fbgemm_fp8.py +38 -138
- transformers/quantizers/quantizer_finegrained_fp8.py +26 -113
- transformers/quantizers/quantizer_fp_quant.py +52 -82
- transformers/quantizers/quantizer_gptq.py +8 -28
- transformers/quantizers/quantizer_higgs.py +42 -60
- transformers/quantizers/quantizer_hqq.py +144 -153
- transformers/quantizers/quantizer_mxfp4.py +14 -194
- transformers/quantizers/quantizer_quanto.py +35 -79
- transformers/quantizers/quantizer_quark.py +36 -17
- transformers/quantizers/quantizer_spqr.py +4 -12
- transformers/quantizers/quantizer_torchao.py +50 -325
- transformers/quantizers/quantizer_vptq.py +4 -27
- transformers/quantizers/quantizers_utils.py +20 -0
- transformers/testing_utils.py +324 -47
- transformers/tokenization_mistral_common.py +7 -2
- transformers/tokenization_utils_base.py +116 -224
- transformers/tokenization_utils_tokenizers.py +190 -106
- transformers/trainer.py +51 -32
- transformers/trainer_callback.py +8 -0
- transformers/trainer_jit_checkpoint.py +126 -0
- transformers/trainer_seq2seq.py +4 -0
- transformers/trainer_utils.py +1 -1
- transformers/training_args.py +74 -38
- transformers/utils/__init__.py +7 -4
- transformers/utils/attention_visualizer.py +4 -4
- transformers/utils/auto_docstring.py +35 -25
- transformers/utils/generic.py +47 -1
- transformers/utils/hub.py +5 -15
- transformers/utils/import_utils.py +112 -25
- transformers/utils/kernel_config.py +74 -19
- transformers/utils/loading_report.py +19 -10
- transformers/utils/quantization_config.py +78 -245
- transformers/video_processing_utils.py +17 -14
- {transformers-5.0.0rc0.dist-info → transformers-5.0.0rc2.dist-info}/METADATA +275 -229
- {transformers-5.0.0rc0.dist-info → transformers-5.0.0rc2.dist-info}/RECORD +832 -777
- {transformers-5.0.0rc0.dist-info → transformers-5.0.0rc2.dist-info}/WHEEL +1 -1
- transformers/kernels/__init__.py +0 -0
- transformers/kernels/falcon_mamba/selective_scan_with_ln_interface.py +0 -529
- transformers/models/roformer/tokenization_roformer_fast.py +0 -160
- {transformers-5.0.0rc0.dist-info → transformers-5.0.0rc2.dist-info}/entry_points.txt +0 -0
- {transformers-5.0.0rc0.dist-info → transformers-5.0.0rc2.dist-info/licenses}/LICENSE +0 -0
- {transformers-5.0.0rc0.dist-info → transformers-5.0.0rc2.dist-info}/top_level.txt +0 -0
|
@@ -1500,6 +1500,7 @@ class AriaForConditionalGeneration(LlavaForConditionalGeneration):
|
|
|
1500
1500
|
attention_mask=None,
|
|
1501
1501
|
cache_position=None,
|
|
1502
1502
|
logits_to_keep=None,
|
|
1503
|
+
is_first_iteration=False,
|
|
1503
1504
|
**kwargs,
|
|
1504
1505
|
):
|
|
1505
1506
|
model_inputs = super().prepare_inputs_for_generation(
|
|
@@ -1509,12 +1510,15 @@ class AriaForConditionalGeneration(LlavaForConditionalGeneration):
|
|
|
1509
1510
|
attention_mask=attention_mask,
|
|
1510
1511
|
cache_position=cache_position,
|
|
1511
1512
|
logits_to_keep=logits_to_keep,
|
|
1513
|
+
is_first_iteration=is_first_iteration,
|
|
1512
1514
|
**kwargs,
|
|
1513
1515
|
)
|
|
1514
1516
|
|
|
1515
|
-
if
|
|
1516
|
-
#
|
|
1517
|
-
#
|
|
1517
|
+
if is_first_iteration or not kwargs.get("use_cache", True):
|
|
1518
|
+
# Pixel values are used only in the first iteration if available
|
|
1519
|
+
# In subsquent iterations, they are already merged with text and cached
|
|
1520
|
+
# NOTE: first iteration doesn't have to be prefill, it can be the first
|
|
1521
|
+
# iteration with a question and cached system prompt (continue generate from cache)
|
|
1518
1522
|
model_inputs["pixel_values"] = pixel_values
|
|
1519
1523
|
model_inputs["pixel_mask"] = pixel_mask
|
|
1520
1524
|
|
|
@@ -32,9 +32,6 @@ if is_torch_available():
|
|
|
32
32
|
|
|
33
33
|
logger = logging.get_logger(__name__)
|
|
34
34
|
|
|
35
|
-
MAX_AUDIO_LEN = 10 * 60 # 10 minutes
|
|
36
|
-
DEFAULT_TRANSCRIPTION_PROMPT = "Transcribe the input speech."
|
|
37
|
-
|
|
38
35
|
|
|
39
36
|
class AudioFlamingo3ProcessorKwargs(ProcessingKwargs, total=False):
|
|
40
37
|
_defaults = {
|
|
@@ -63,32 +60,41 @@ class AudioFlamingo3Processor(ProcessorMixin):
|
|
|
63
60
|
[`Qwen2TokenizerFast`]. See the [`~AudioFlamingo3Processor.__call__`] for more information.
|
|
64
61
|
|
|
65
62
|
Args:
|
|
66
|
-
|
|
67
|
-
|
|
68
|
-
|
|
69
|
-
|
|
70
|
-
|
|
71
|
-
|
|
72
|
-
|
|
73
|
-
|
|
74
|
-
|
|
63
|
+
feature_extractor ([`WhisperFeatureExtractor`]):
|
|
64
|
+
The feature extractor is a required input.
|
|
65
|
+
tokenizer ([`Qwen2TokenizerFast`]):
|
|
66
|
+
The tokenizer is a required input.
|
|
67
|
+
chat_template (`Optional[str]`, *optional*):
|
|
68
|
+
The Jinja template to use for formatting the conversation. If not provided, the tokenizer's default chat
|
|
69
|
+
template will be used.
|
|
70
|
+
audio_token (`Optional[str]`, *optional*, defaults to `"<sound>"`):
|
|
71
|
+
Special token used to represent audio inputs in the chat template.
|
|
72
|
+
default_transcription_prompt (`str`, *optional*, defaults to `"Transcribe the input speech."`):
|
|
73
|
+
Default prompt to use for transcription tasks when applying transcription requests.
|
|
74
|
+
max_audio_len (`int`, *optional*, defaults to 600):
|
|
75
|
+
Maximum length of audio sequences in seconds. Audio longer than this will be truncated.
|
|
75
76
|
"""
|
|
76
77
|
|
|
77
|
-
attributes = ["feature_extractor", "tokenizer"]
|
|
78
|
-
feature_extractor_class = "WhisperFeatureExtractor"
|
|
79
|
-
tokenizer_class = "Qwen2TokenizerFast"
|
|
80
|
-
|
|
81
78
|
def __init__(
|
|
82
79
|
self,
|
|
83
80
|
feature_extractor,
|
|
84
81
|
tokenizer,
|
|
85
82
|
chat_template=None,
|
|
86
83
|
audio_token="<sound>",
|
|
84
|
+
default_transcription_prompt="Transcribe the input speech.",
|
|
85
|
+
max_audio_len=600,
|
|
87
86
|
):
|
|
88
87
|
self.audio_token = audio_token
|
|
89
88
|
self.audio_token_id = tokenizer.convert_tokens_to_ids(audio_token)
|
|
89
|
+
self.default_transcription_prompt = default_transcription_prompt
|
|
90
|
+
self.max_audio_len = max_audio_len
|
|
90
91
|
super().__init__(feature_extractor, tokenizer, chat_template=chat_template)
|
|
91
92
|
|
|
93
|
+
def _get_audio_token_length(self, audio_lengths: "torch.Tensor") -> "torch.Tensor":
|
|
94
|
+
conv_output_lengths = (audio_lengths - 1) // 2 + 1 # After conv2 downsampling
|
|
95
|
+
audio_tokens_lengths = (conv_output_lengths - 2) // 2 + 1 # After avg pooling
|
|
96
|
+
return audio_tokens_lengths
|
|
97
|
+
|
|
92
98
|
def __call__(
|
|
93
99
|
self,
|
|
94
100
|
text: Union[TextInput, list[TextInput]],
|
|
@@ -143,7 +149,7 @@ class AudioFlamingo3Processor(ProcessorMixin):
|
|
|
143
149
|
|
|
144
150
|
# Determine number of chunks per sample, and flatten
|
|
145
151
|
window_size = int(audio_kwargs["sampling_rate"] * audio_kwargs["chunk_length"])
|
|
146
|
-
max_windows = int(
|
|
152
|
+
max_windows = int(self.max_audio_len // audio_kwargs["chunk_length"])
|
|
147
153
|
|
|
148
154
|
per_sample_windows: list[int] = []
|
|
149
155
|
flat_chunks: list[np.ndarray] = []
|
|
@@ -153,7 +159,7 @@ class AudioFlamingo3Processor(ProcessorMixin):
|
|
|
153
159
|
n_win = max(1, (n_samples + window_size - 1) // window_size)
|
|
154
160
|
if n_win > max_windows:
|
|
155
161
|
logger.warning(
|
|
156
|
-
f"Audio duration ({n_samples / audio_kwargs['sampling_rate']:.1f}s) exceeds {
|
|
162
|
+
f"Audio duration ({n_samples / audio_kwargs['sampling_rate']:.1f}s) exceeds {self.max_audio_len}s; truncating to first {self.max_audio_len}s."
|
|
157
163
|
)
|
|
158
164
|
n_win = max_windows
|
|
159
165
|
per_sample_windows.append(n_win)
|
|
@@ -171,8 +177,7 @@ class AudioFlamingo3Processor(ProcessorMixin):
|
|
|
171
177
|
|
|
172
178
|
# Compute sequence lengths token counting
|
|
173
179
|
audio_lengths = torch.stack([s.sum() for s in torch.split(padding_mask.sum(-1), per_sample_windows)])
|
|
174
|
-
|
|
175
|
-
audio_tokens_lengths = (conv_output_lengths - 2) // 2 + 1 # After avg pooling
|
|
180
|
+
audio_tokens_lengths = self._get_audio_token_length(audio_lengths)
|
|
176
181
|
|
|
177
182
|
# expand audio tokens in text
|
|
178
183
|
for i, audio_length in enumerate(audio_tokens_lengths):
|
|
@@ -236,7 +241,7 @@ class AudioFlamingo3Processor(ProcessorMixin):
|
|
|
236
241
|
raise ValueError("`audio` must contain at least one sample.")
|
|
237
242
|
|
|
238
243
|
if prompt is None:
|
|
239
|
-
prompts = [
|
|
244
|
+
prompts = [self.default_transcription_prompt] * batch_size
|
|
240
245
|
elif isinstance(prompt, str):
|
|
241
246
|
prompts = [prompt] * batch_size
|
|
242
247
|
elif isinstance(prompt, (list, tuple)):
|
|
@@ -247,7 +252,7 @@ class AudioFlamingo3Processor(ProcessorMixin):
|
|
|
247
252
|
prompts = []
|
|
248
253
|
for item in prompt:
|
|
249
254
|
if item is None:
|
|
250
|
-
prompts.append(
|
|
255
|
+
prompts.append(self.default_transcription_prompt)
|
|
251
256
|
elif isinstance(item, str):
|
|
252
257
|
prompts.append(item)
|
|
253
258
|
else:
|
|
@@ -543,7 +543,7 @@ def add_generation_mixin_to_remote_model(model_class):
|
|
|
543
543
|
|
|
544
544
|
class _LazyAutoMapping(OrderedDict[type[PreTrainedConfig], _LazyAutoMappingValue]):
|
|
545
545
|
"""
|
|
546
|
-
|
|
546
|
+
A mapping config to object (model or tokenizer for instance) that will load keys and values when it is accessed.
|
|
547
547
|
|
|
548
548
|
Args:
|
|
549
549
|
- config_mapping: The map model type to config class
|
|
@@ -142,12 +142,14 @@ CONFIG_MAPPING_NAMES = OrderedDict[str, str](
|
|
|
142
142
|
("ernie", "ErnieConfig"),
|
|
143
143
|
("ernie4_5", "Ernie4_5Config"),
|
|
144
144
|
("ernie4_5_moe", "Ernie4_5_MoeConfig"),
|
|
145
|
+
("ernie4_5_vl_moe", "Ernie4_5_VL_MoeConfig"),
|
|
145
146
|
("esm", "EsmConfig"),
|
|
146
147
|
("evolla", "EvollaConfig"),
|
|
147
148
|
("exaone4", "Exaone4Config"),
|
|
148
149
|
("falcon", "FalconConfig"),
|
|
149
150
|
("falcon_h1", "FalconH1Config"),
|
|
150
151
|
("falcon_mamba", "FalconMambaConfig"),
|
|
152
|
+
("fast_vlm", "FastVlmConfig"),
|
|
151
153
|
("fastspeech2_conformer", "FastSpeech2ConformerConfig"),
|
|
152
154
|
("fastspeech2_conformer_with_hifigan", "FastSpeech2ConformerWithHifiGanConfig"),
|
|
153
155
|
("flaubert", "FlaubertConfig"),
|
|
@@ -178,6 +180,8 @@ CONFIG_MAPPING_NAMES = OrderedDict[str, str](
|
|
|
178
180
|
("glm4v_moe_vision", "Glm4vMoeVisionConfig"),
|
|
179
181
|
("glm4v_text", "Glm4vTextConfig"),
|
|
180
182
|
("glm4v_vision", "Glm4vVisionConfig"),
|
|
183
|
+
("glmasr", "GlmAsrConfig"),
|
|
184
|
+
("glmasr_encoder", "GlmAsrEncoderConfig"),
|
|
181
185
|
("glpn", "GLPNConfig"),
|
|
182
186
|
("got_ocr2", "GotOcr2Config"),
|
|
183
187
|
("gpt-sw3", "GPT2Config"),
|
|
@@ -214,12 +218,15 @@ CONFIG_MAPPING_NAMES = OrderedDict[str, str](
|
|
|
214
218
|
("instructblipvideo", "InstructBlipVideoConfig"),
|
|
215
219
|
("internvl", "InternVLConfig"),
|
|
216
220
|
("internvl_vision", "InternVLVisionConfig"),
|
|
221
|
+
("jais2", "Jais2Config"),
|
|
217
222
|
("jamba", "JambaConfig"),
|
|
218
223
|
("janus", "JanusConfig"),
|
|
219
224
|
("jetmoe", "JetMoeConfig"),
|
|
220
225
|
("kosmos-2", "Kosmos2Config"),
|
|
221
226
|
("kosmos-2.5", "Kosmos2_5Config"),
|
|
222
227
|
("kyutai_speech_to_text", "KyutaiSpeechToTextConfig"),
|
|
228
|
+
("lasr_ctc", "LasrCTCConfig"),
|
|
229
|
+
("lasr_encoder", "LasrEncoderConfig"),
|
|
223
230
|
("layoutlm", "LayoutLMConfig"),
|
|
224
231
|
("layoutlmv2", "LayoutLMv2Config"),
|
|
225
232
|
("layoutlmv3", "LayoutLMv3Config"),
|
|
@@ -297,11 +304,18 @@ CONFIG_MAPPING_NAMES = OrderedDict[str, str](
|
|
|
297
304
|
("ovis2", "Ovis2Config"),
|
|
298
305
|
("owlv2", "Owlv2Config"),
|
|
299
306
|
("owlvit", "OwlViTConfig"),
|
|
307
|
+
("paddleocr_vl", "PaddleOCRVLConfig"),
|
|
300
308
|
("paligemma", "PaliGemmaConfig"),
|
|
301
309
|
("parakeet_ctc", "ParakeetCTCConfig"),
|
|
302
310
|
("parakeet_encoder", "ParakeetEncoderConfig"),
|
|
303
311
|
("patchtsmixer", "PatchTSMixerConfig"),
|
|
304
312
|
("patchtst", "PatchTSTConfig"),
|
|
313
|
+
("pe_audio", "PeAudioConfig"),
|
|
314
|
+
("pe_audio_encoder", "PeAudioEncoderConfig"),
|
|
315
|
+
("pe_audio_video", "PeAudioVideoConfig"),
|
|
316
|
+
("pe_audio_video_encoder", "PeAudioVideoEncoderConfig"),
|
|
317
|
+
("pe_video", "PeVideoConfig"),
|
|
318
|
+
("pe_video_encoder", "PeVideoEncoderConfig"),
|
|
305
319
|
("pegasus", "PegasusConfig"),
|
|
306
320
|
("pegasus_x", "PegasusXConfig"),
|
|
307
321
|
("perceiver", "PerceiverConfig"),
|
|
@@ -312,6 +326,7 @@ CONFIG_MAPPING_NAMES = OrderedDict[str, str](
|
|
|
312
326
|
("phi4_multimodal", "Phi4MultimodalConfig"),
|
|
313
327
|
("phimoe", "PhimoeConfig"),
|
|
314
328
|
("pix2struct", "Pix2StructConfig"),
|
|
329
|
+
("pixio", "PixioConfig"),
|
|
315
330
|
("pixtral", "PixtralVisionConfig"),
|
|
316
331
|
("plbart", "PLBartConfig"),
|
|
317
332
|
("poolformer", "PoolFormerConfig"),
|
|
@@ -578,6 +593,7 @@ MODEL_NAMES_MAPPING = OrderedDict[str, str](
|
|
|
578
593
|
("ernie", "ERNIE"),
|
|
579
594
|
("ernie4_5", "Ernie4_5"),
|
|
580
595
|
("ernie4_5_moe", "Ernie4_5_MoE"),
|
|
596
|
+
("ernie4_5_vl_moe", "Ernie4_5_VL_MoE"),
|
|
581
597
|
("esm", "ESM"),
|
|
582
598
|
("evolla", "Evolla"),
|
|
583
599
|
("exaone4", "EXAONE-4.0"),
|
|
@@ -585,6 +601,7 @@ MODEL_NAMES_MAPPING = OrderedDict[str, str](
|
|
|
585
601
|
("falcon3", "Falcon3"),
|
|
586
602
|
("falcon_h1", "FalconH1"),
|
|
587
603
|
("falcon_mamba", "FalconMamba"),
|
|
604
|
+
("fast_vlm", "FastVlm"),
|
|
588
605
|
("fastspeech2_conformer", "FastSpeech2Conformer"),
|
|
589
606
|
("fastspeech2_conformer_with_hifigan", "FastSpeech2ConformerWithHifiGan"),
|
|
590
607
|
("flan-t5", "FLAN-T5"),
|
|
@@ -617,6 +634,8 @@ MODEL_NAMES_MAPPING = OrderedDict[str, str](
|
|
|
617
634
|
("glm4v_moe_vision", "Glm4vMoeVisionModel"),
|
|
618
635
|
("glm4v_text", "GLM4V"),
|
|
619
636
|
("glm4v_vision", "Glm4vVisionModel"),
|
|
637
|
+
("glmasr", "GLM-ASR"),
|
|
638
|
+
("glmasr_encoder", "GLM-ASR Encoder"),
|
|
620
639
|
("glpn", "GLPN"),
|
|
621
640
|
("got_ocr2", "GOT-OCR2"),
|
|
622
641
|
("gpt-sw3", "GPT-Sw3"),
|
|
@@ -654,12 +673,16 @@ MODEL_NAMES_MAPPING = OrderedDict[str, str](
|
|
|
654
673
|
("instructblipvideo", "InstructBlipVideo"),
|
|
655
674
|
("internvl", "InternVL"),
|
|
656
675
|
("internvl_vision", "InternVLVision"),
|
|
676
|
+
("jais2", "Jais2"),
|
|
657
677
|
("jamba", "Jamba"),
|
|
658
678
|
("janus", "Janus"),
|
|
659
679
|
("jetmoe", "JetMoe"),
|
|
660
680
|
("kosmos-2", "KOSMOS-2"),
|
|
661
681
|
("kosmos-2.5", "KOSMOS-2.5"),
|
|
662
682
|
("kyutai_speech_to_text", "KyutaiSpeechToText"),
|
|
683
|
+
("lasr", "Lasr"),
|
|
684
|
+
("lasr_ctc", "Lasr"),
|
|
685
|
+
("lasr_encoder", "LasrEncoder"),
|
|
663
686
|
("layoutlm", "LayoutLM"),
|
|
664
687
|
("layoutlmv2", "LayoutLMv2"),
|
|
665
688
|
("layoutlmv3", "LayoutLMv3"),
|
|
@@ -747,12 +770,19 @@ MODEL_NAMES_MAPPING = OrderedDict[str, str](
|
|
|
747
770
|
("ovis2", "Ovis2"),
|
|
748
771
|
("owlv2", "OWLv2"),
|
|
749
772
|
("owlvit", "OWL-ViT"),
|
|
773
|
+
("paddleocr_vl", "PaddleOCRVL"),
|
|
750
774
|
("paligemma", "PaliGemma"),
|
|
751
775
|
("parakeet", "Parakeet"),
|
|
752
776
|
("parakeet_ctc", "Parakeet"),
|
|
753
777
|
("parakeet_encoder", "ParakeetEncoder"),
|
|
754
778
|
("patchtsmixer", "PatchTSMixer"),
|
|
755
779
|
("patchtst", "PatchTST"),
|
|
780
|
+
("pe_audio", "PeAudio"),
|
|
781
|
+
("pe_audio_encoder", "PeAudioEncoder"),
|
|
782
|
+
("pe_audio_video", "PeAudioVideo"),
|
|
783
|
+
("pe_audio_video_encoder", "PeAudioVideoEncoder"),
|
|
784
|
+
("pe_video", "PeVideo"),
|
|
785
|
+
("pe_video_encoder", "PeVideoEncoder"),
|
|
756
786
|
("pegasus", "Pegasus"),
|
|
757
787
|
("pegasus_x", "PEGASUS-X"),
|
|
758
788
|
("perceiver", "Perceiver"),
|
|
@@ -764,6 +794,7 @@ MODEL_NAMES_MAPPING = OrderedDict[str, str](
|
|
|
764
794
|
("phimoe", "Phimoe"),
|
|
765
795
|
("phobert", "PhoBERT"),
|
|
766
796
|
("pix2struct", "Pix2Struct"),
|
|
797
|
+
("pixio", "Pixio"),
|
|
767
798
|
("pixtral", "Pixtral"),
|
|
768
799
|
("plbart", "PLBart"),
|
|
769
800
|
("poolformer", "PoolFormer"),
|
|
@@ -946,6 +977,7 @@ SPECIAL_MODEL_TYPE_TO_MODULE_NAME = OrderedDict[str, str](
|
|
|
946
977
|
("glm4v_moe_vision", "glm4v_moe"),
|
|
947
978
|
("glm4v_text", "glm4v"),
|
|
948
979
|
("glm4v_moe_text", "glm4v_moe"),
|
|
980
|
+
("glmasr_encoder", "glmasr"),
|
|
949
981
|
("grounding-dino", "grounding_dino"),
|
|
950
982
|
("mm-grounding-dino", "mm_grounding_dino"),
|
|
951
983
|
("idefics3_vision", "idefics3"),
|
|
@@ -972,9 +1004,15 @@ SPECIAL_MODEL_TYPE_TO_MODULE_NAME = OrderedDict[str, str](
|
|
|
972
1004
|
("llama4_text", "llama4"),
|
|
973
1005
|
("blip_2_qformer", "blip_2"),
|
|
974
1006
|
("fastspeech2_conformer_with_hifigan", "fastspeech2_conformer"),
|
|
1007
|
+
("perception_encoder", "perception_lm"),
|
|
1008
|
+
("pe_audio_encoder", "pe_audio"),
|
|
1009
|
+
("pe_video_encoder", "pe_video"),
|
|
1010
|
+
("pe_audio_video_encoder", "pe_audio_video"),
|
|
975
1011
|
("video_llama_3_vision", "video_llama_3"),
|
|
976
1012
|
("parakeet_encoder", "parakeet"),
|
|
977
1013
|
("parakeet_ctc", "parakeet"),
|
|
1014
|
+
("lasr_encoder", "lasr"),
|
|
1015
|
+
("lasr_ctc", "lasr"),
|
|
978
1016
|
("wav2vec2-bert", "wav2vec2_bert"),
|
|
979
1017
|
]
|
|
980
1018
|
)
|
|
@@ -38,6 +38,7 @@ logger = logging.get_logger(__name__)
|
|
|
38
38
|
FEATURE_EXTRACTOR_MAPPING_NAMES = OrderedDict(
|
|
39
39
|
[
|
|
40
40
|
("audio-spectrogram-transformer", "ASTFeatureExtractor"),
|
|
41
|
+
("audioflamingo3", "WhisperFeatureExtractor"),
|
|
41
42
|
("clap", "ClapFeatureExtractor"),
|
|
42
43
|
("clvp", "ClvpFeatureExtractor"),
|
|
43
44
|
("csm", "EncodecFeatureExtractor"),
|
|
@@ -46,9 +47,12 @@ FEATURE_EXTRACTOR_MAPPING_NAMES = OrderedDict(
|
|
|
46
47
|
("dia", "DiaFeatureExtractor"),
|
|
47
48
|
("encodec", "EncodecFeatureExtractor"),
|
|
48
49
|
("gemma3n", "Gemma3nAudioFeatureExtractor"),
|
|
50
|
+
("glmasr", "WhisperFeatureExtractor"),
|
|
49
51
|
("granite_speech", "GraniteSpeechFeatureExtractor"),
|
|
50
52
|
("hubert", "Wav2Vec2FeatureExtractor"),
|
|
51
53
|
("kyutai_speech_to_text", "KyutaiSpeechToTextFeatureExtractor"),
|
|
54
|
+
("lasr_ctc", "LasrFeatureExtractor"),
|
|
55
|
+
("lasr_encoder", "LasrFeatureExtractor"),
|
|
52
56
|
("markuplm", "MarkupLMFeatureExtractor"),
|
|
53
57
|
("mimi", "EncodecFeatureExtractor"),
|
|
54
58
|
("moonshine", "Wav2Vec2FeatureExtractor"),
|
|
@@ -57,6 +61,8 @@ FEATURE_EXTRACTOR_MAPPING_NAMES = OrderedDict(
|
|
|
57
61
|
("musicgen_melody", "MusicgenMelodyFeatureExtractor"),
|
|
58
62
|
("parakeet_ctc", "ParakeetFeatureExtractor"),
|
|
59
63
|
("parakeet_encoder", "ParakeetFeatureExtractor"),
|
|
64
|
+
("pe_audio", "PeAudioFeatureExtractor"),
|
|
65
|
+
("pe_audio_video", "PeAudioFeatureExtractor"),
|
|
60
66
|
("phi4_multimodal", "Phi4MultimodalFeatureExtractor"),
|
|
61
67
|
("pop2piano", "Pop2PianoFeatureExtractor"),
|
|
62
68
|
("qwen2_5_omni", "WhisperFeatureExtractor"),
|
|
@@ -343,13 +349,13 @@ class AutoFeatureExtractor:
|
|
|
343
349
|
)
|
|
344
350
|
_ = kwargs.pop("code_revision", None)
|
|
345
351
|
feature_extractor_class.register_for_auto_class()
|
|
346
|
-
return feature_extractor_class.
|
|
352
|
+
return feature_extractor_class.from_pretrained(pretrained_model_name_or_path, **kwargs)
|
|
347
353
|
elif feature_extractor_class is not None:
|
|
348
|
-
return feature_extractor_class.
|
|
354
|
+
return feature_extractor_class.from_pretrained(pretrained_model_name_or_path, **kwargs)
|
|
349
355
|
# Last try: we use the FEATURE_EXTRACTOR_MAPPING.
|
|
350
356
|
elif type(config) in FEATURE_EXTRACTOR_MAPPING:
|
|
351
357
|
feature_extractor_class = FEATURE_EXTRACTOR_MAPPING[type(config)]
|
|
352
|
-
return feature_extractor_class.
|
|
358
|
+
return feature_extractor_class.from_pretrained(pretrained_model_name_or_path, **kwargs)
|
|
353
359
|
|
|
354
360
|
raise ValueError(
|
|
355
361
|
f"Unrecognized feature extractor in {pretrained_model_name_or_path}. Should have a "
|
|
@@ -100,6 +100,7 @@ else:
|
|
|
100
100
|
("efficientnet", ("EfficientNetImageProcessor", "EfficientNetImageProcessorFast")),
|
|
101
101
|
("emu3", ("Emu3ImageProcessor", None)),
|
|
102
102
|
("eomt", ("EomtImageProcessor", "EomtImageProcessorFast")),
|
|
103
|
+
("ernie4_5_vl_moe", ("Ernie4_5_VL_MoeImageProcessor", "Ernie4_5_VL_MoeImageProcessorFast")),
|
|
103
104
|
("flava", ("FlavaImageProcessor", "FlavaImageProcessorFast")),
|
|
104
105
|
("florence2", ("CLIPImageProcessor", "CLIPImageProcessorFast")),
|
|
105
106
|
("focalnet", ("BitImageProcessor", "BitImageProcessorFast")),
|
|
@@ -153,11 +154,13 @@ else:
|
|
|
153
154
|
("ovis2", ("Ovis2ImageProcessor", "Ovis2ImageProcessorFast")),
|
|
154
155
|
("owlv2", ("Owlv2ImageProcessor", "Owlv2ImageProcessorFast")),
|
|
155
156
|
("owlvit", ("OwlViTImageProcessor", "OwlViTImageProcessorFast")),
|
|
157
|
+
("paddleocr_vl", ("PaddleOCRVLImageProcessor", "PaddleOCRVLImageProcessorFast")),
|
|
156
158
|
("paligemma", ("SiglipImageProcessor", "SiglipImageProcessorFast")),
|
|
157
159
|
("perceiver", ("PerceiverImageProcessor", "PerceiverImageProcessorFast")),
|
|
158
160
|
("perception_lm", (None, "PerceptionLMImageProcessorFast")),
|
|
159
161
|
("phi4_multimodal", (None, "Phi4MultimodalImageProcessorFast")),
|
|
160
162
|
("pix2struct", ("Pix2StructImageProcessor", "Pix2StructImageProcessorFast")),
|
|
163
|
+
("pixio", ("BitImageProcessor", "BitImageProcessorFast")),
|
|
161
164
|
("pixtral", ("PixtralImageProcessor", "PixtralImageProcessorFast")),
|
|
162
165
|
("poolformer", ("PoolFormerImageProcessor", "PoolFormerImageProcessorFast")),
|
|
163
166
|
("prompt_depth_anything", ("PromptDepthAnythingImageProcessor", "PromptDepthAnythingImageProcessorFast")),
|
|
@@ -604,9 +607,9 @@ class AutoImageProcessor:
|
|
|
604
607
|
image_processor_class = get_class_from_dynamic_module(class_ref, pretrained_model_name_or_path, **kwargs)
|
|
605
608
|
_ = kwargs.pop("code_revision", None)
|
|
606
609
|
image_processor_class.register_for_auto_class()
|
|
607
|
-
return image_processor_class.
|
|
610
|
+
return image_processor_class.from_pretrained(pretrained_model_name_or_path, *inputs, **kwargs)
|
|
608
611
|
elif image_processor_class is not None:
|
|
609
|
-
return image_processor_class.
|
|
612
|
+
return image_processor_class.from_pretrained(pretrained_model_name_or_path, *inputs, **kwargs)
|
|
610
613
|
# Last try: we use the IMAGE_PROCESSOR_MAPPING.
|
|
611
614
|
elif type(config) in IMAGE_PROCESSOR_MAPPING:
|
|
612
615
|
image_processor_tuple = IMAGE_PROCESSOR_MAPPING[type(config)]
|
|
@@ -145,12 +145,14 @@ MODEL_MAPPING_NAMES = OrderedDict(
|
|
|
145
145
|
("ernie", "ErnieModel"),
|
|
146
146
|
("ernie4_5", "Ernie4_5Model"),
|
|
147
147
|
("ernie4_5_moe", "Ernie4_5_MoeModel"),
|
|
148
|
+
("ernie4_5_vl_moe", "Ernie4_5_VL_MoeModel"),
|
|
148
149
|
("esm", "EsmModel"),
|
|
149
150
|
("evolla", "EvollaModel"),
|
|
150
151
|
("exaone4", "Exaone4Model"),
|
|
151
152
|
("falcon", "FalconModel"),
|
|
152
153
|
("falcon_h1", "FalconH1Model"),
|
|
153
154
|
("falcon_mamba", "FalconMambaModel"),
|
|
155
|
+
("fast_vlm", "FastVlmModel"),
|
|
154
156
|
("fastspeech2_conformer", "FastSpeech2ConformerModel"),
|
|
155
157
|
("fastspeech2_conformer_with_hifigan", "FastSpeech2ConformerWithHifiGan"),
|
|
156
158
|
("flaubert", "FlaubertModel"),
|
|
@@ -181,6 +183,8 @@ MODEL_MAPPING_NAMES = OrderedDict(
|
|
|
181
183
|
("glm4v_moe_vision", "Glm4vMoeVisionModel"),
|
|
182
184
|
("glm4v_text", "Glm4vTextModel"),
|
|
183
185
|
("glm4v_vision", "Glm4vVisionModel"),
|
|
186
|
+
("glmasr", "GlmAsrForConditionalGeneration"),
|
|
187
|
+
("glmasr_encoder", "GlmAsrEncoder"),
|
|
184
188
|
("glpn", "GLPNModel"),
|
|
185
189
|
("got_ocr2", "GotOcr2Model"),
|
|
186
190
|
("gpt-sw3", "GPT2Model"),
|
|
@@ -215,12 +219,15 @@ MODEL_MAPPING_NAMES = OrderedDict(
|
|
|
215
219
|
("instructblipvideo", "InstructBlipVideoModel"),
|
|
216
220
|
("internvl", "InternVLModel"),
|
|
217
221
|
("internvl_vision", "InternVLVisionModel"),
|
|
222
|
+
("jais2", "Jais2Model"),
|
|
218
223
|
("jamba", "JambaModel"),
|
|
219
224
|
("janus", "JanusModel"),
|
|
220
225
|
("jetmoe", "JetMoeModel"),
|
|
221
226
|
("kosmos-2", "Kosmos2Model"),
|
|
222
227
|
("kosmos-2.5", "Kosmos2_5Model"),
|
|
223
228
|
("kyutai_speech_to_text", "KyutaiSpeechToTextModel"),
|
|
229
|
+
("lasr_ctc", "LasrForCTC"),
|
|
230
|
+
("lasr_encoder", "LasrEncoder"),
|
|
224
231
|
("layoutlm", "LayoutLMModel"),
|
|
225
232
|
("layoutlmv2", "LayoutLMv2Model"),
|
|
226
233
|
("layoutlmv3", "LayoutLMv3Model"),
|
|
@@ -301,6 +308,12 @@ MODEL_MAPPING_NAMES = OrderedDict(
|
|
|
301
308
|
("parakeet_encoder", "ParakeetEncoder"),
|
|
302
309
|
("patchtsmixer", "PatchTSMixerModel"),
|
|
303
310
|
("patchtst", "PatchTSTModel"),
|
|
311
|
+
("pe_audio", "PeAudioModel"),
|
|
312
|
+
("pe_audio_encoder", "PeAudioEncoder"),
|
|
313
|
+
("pe_audio_video", "PeAudioVideoModel"),
|
|
314
|
+
("pe_audio_video_encoder", "PeAudioVideoEncoder"),
|
|
315
|
+
("pe_video", "PeVideoModel"),
|
|
316
|
+
("pe_video_encoder", "PeVideoEncoder"),
|
|
304
317
|
("pegasus", "PegasusModel"),
|
|
305
318
|
("pegasus_x", "PegasusXModel"),
|
|
306
319
|
("perceiver", "PerceiverModel"),
|
|
@@ -310,6 +323,7 @@ MODEL_MAPPING_NAMES = OrderedDict(
|
|
|
310
323
|
("phi3", "Phi3Model"),
|
|
311
324
|
("phi4_multimodal", "Phi4MultimodalModel"),
|
|
312
325
|
("phimoe", "PhimoeModel"),
|
|
326
|
+
("pixio", "PixioModel"),
|
|
313
327
|
("pixtral", "PixtralVisionModel"),
|
|
314
328
|
("plbart", "PLBartModel"),
|
|
315
329
|
("poolformer", "PoolFormerModel"),
|
|
@@ -467,6 +481,7 @@ MODEL_FOR_PRETRAINING_MAPPING_NAMES = OrderedDict(
|
|
|
467
481
|
("fsmt", "FSMTForConditionalGeneration"),
|
|
468
482
|
("funnel", "FunnelForPreTraining"),
|
|
469
483
|
("gemma3", "Gemma3ForConditionalGeneration"),
|
|
484
|
+
("glmasr", "GlmAsrForConditionalGeneration"),
|
|
470
485
|
("gpt-sw3", "GPT2LMHeadModel"),
|
|
471
486
|
("gpt2", "GPT2LMHeadModel"),
|
|
472
487
|
("gpt_bigcode", "GPTBigCodeForCausalLM"),
|
|
@@ -686,6 +701,7 @@ MODEL_FOR_CAUSAL_LM_MAPPING_NAMES = OrderedDict(
|
|
|
686
701
|
("helium", "HeliumForCausalLM"),
|
|
687
702
|
("hunyuan_v1_dense", "HunYuanDenseV1ForCausalLM"),
|
|
688
703
|
("hunyuan_v1_moe", "HunYuanMoEV1ForCausalLM"),
|
|
704
|
+
("jais2", "Jais2ForCausalLM"),
|
|
689
705
|
("jamba", "JambaForCausalLM"),
|
|
690
706
|
("jetmoe", "JetMoeForCausalLM"),
|
|
691
707
|
("lfm2", "Lfm2ForCausalLM"),
|
|
@@ -795,6 +811,7 @@ MODEL_FOR_IMAGE_MAPPING_NAMES = OrderedDict(
|
|
|
795
811
|
("mobilenet_v2", "MobileNetV2Model"),
|
|
796
812
|
("mobilevit", "MobileViTModel"),
|
|
797
813
|
("mobilevitv2", "MobileViTV2Model"),
|
|
814
|
+
("pixio", "PixioModel"),
|
|
798
815
|
("poolformer", "PoolFormerModel"),
|
|
799
816
|
("pvt", "PvtModel"),
|
|
800
817
|
("regnet", "RegNetModel"),
|
|
@@ -995,7 +1012,9 @@ MODEL_FOR_IMAGE_TEXT_TO_TEXT_MAPPING_NAMES = OrderedDict(
|
|
|
995
1012
|
("deepseek_vl", "DeepseekVLForConditionalGeneration"),
|
|
996
1013
|
("deepseek_vl_hybrid", "DeepseekVLHybridForConditionalGeneration"),
|
|
997
1014
|
("emu3", "Emu3ForConditionalGeneration"),
|
|
1015
|
+
("ernie4_5_vl_moe", "Ernie4_5_VL_MoeForConditionalGeneration"),
|
|
998
1016
|
("evolla", "EvollaForProteinText2Text"),
|
|
1017
|
+
("fast_vlm", "FastVlmForConditionalGeneration"),
|
|
999
1018
|
("florence2", "Florence2ForConditionalGeneration"),
|
|
1000
1019
|
("fuyu", "FuyuForCausalLM"),
|
|
1001
1020
|
("gemma3", "Gemma3ForConditionalGeneration"),
|
|
@@ -1022,6 +1041,7 @@ MODEL_FOR_IMAGE_TEXT_TO_TEXT_MAPPING_NAMES = OrderedDict(
|
|
|
1022
1041
|
("mistral3", "Mistral3ForConditionalGeneration"),
|
|
1023
1042
|
("mllama", "MllamaForConditionalGeneration"),
|
|
1024
1043
|
("ovis2", "Ovis2ForConditionalGeneration"),
|
|
1044
|
+
("paddleocr_vl", "PaddleOCRVLForConditionalGeneration"),
|
|
1025
1045
|
("paligemma", "PaliGemmaForConditionalGeneration"),
|
|
1026
1046
|
("perception_lm", "PerceptionLMForConditionalGeneration"),
|
|
1027
1047
|
("pix2struct", "Pix2StructForConditionalGeneration"),
|
|
@@ -1045,6 +1065,7 @@ MODEL_FOR_IMAGE_TEXT_TO_TEXT_MAPPING_NAMES = OrderedDict(
|
|
|
1045
1065
|
MODEL_FOR_MULTIMODAL_LM_MAPPING_NAMES = OrderedDict(
|
|
1046
1066
|
[
|
|
1047
1067
|
*list(MODEL_FOR_IMAGE_TEXT_TO_TEXT_MAPPING_NAMES.items()),
|
|
1068
|
+
("glmasr", "GlmAsrForConditionalGeneration"),
|
|
1048
1069
|
("granite_speech", "GraniteSpeechForConditionalGeneration"),
|
|
1049
1070
|
("kyutai_speech_to_text", "KyutaiSpeechToTextForConditionalGeneration"),
|
|
1050
1071
|
("phi4_multimodal", "Phi4MultimodalForCausalLM"),
|
|
@@ -1055,6 +1076,13 @@ MODEL_FOR_MULTIMODAL_LM_MAPPING_NAMES = OrderedDict(
|
|
|
1055
1076
|
]
|
|
1056
1077
|
)
|
|
1057
1078
|
|
|
1079
|
+
MODEL_FOR_MASKED_LM_MAPPING_NAMES = OrderedDict(
|
|
1080
|
+
[
|
|
1081
|
+
# Model for Masked LM mapping
|
|
1082
|
+
("albert", "AlbertForMaskedLM"),
|
|
1083
|
+
]
|
|
1084
|
+
)
|
|
1085
|
+
|
|
1058
1086
|
MODEL_FOR_MASKED_LM_MAPPING_NAMES = OrderedDict(
|
|
1059
1087
|
[
|
|
1060
1088
|
# Model for Masked LM mapping
|
|
@@ -1151,6 +1179,7 @@ MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING_NAMES = OrderedDict(
|
|
|
1151
1179
|
("blenderbot-small", "BlenderbotSmallForConditionalGeneration"),
|
|
1152
1180
|
("encoder-decoder", "EncoderDecoderModel"),
|
|
1153
1181
|
("fsmt", "FSMTForConditionalGeneration"),
|
|
1182
|
+
("glmasr", "GlmAsrForConditionalGeneration"),
|
|
1154
1183
|
("granite_speech", "GraniteSpeechForConditionalGeneration"),
|
|
1155
1184
|
("led", "LEDForConditionalGeneration"),
|
|
1156
1185
|
("longt5", "LongT5ForConditionalGeneration"),
|
|
@@ -1176,6 +1205,12 @@ MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING_NAMES = OrderedDict(
|
|
|
1176
1205
|
]
|
|
1177
1206
|
)
|
|
1178
1207
|
|
|
1208
|
+
MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPING_NAMES = OrderedDict(
|
|
1209
|
+
[
|
|
1210
|
+
("dia", "DiaForConditionalGeneration"),
|
|
1211
|
+
]
|
|
1212
|
+
)
|
|
1213
|
+
|
|
1179
1214
|
MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPING_NAMES = OrderedDict(
|
|
1180
1215
|
[
|
|
1181
1216
|
("dia", "DiaForConditionalGeneration"),
|
|
@@ -1581,6 +1616,7 @@ MODEL_FOR_CTC_MAPPING_NAMES = OrderedDict(
|
|
|
1581
1616
|
# Model for Connectionist temporal classification (CTC) mapping
|
|
1582
1617
|
("data2vec-audio", "Data2VecAudioForCTC"),
|
|
1583
1618
|
("hubert", "HubertForCTC"),
|
|
1619
|
+
("lasr_ctc", "LasrForCTC"),
|
|
1584
1620
|
("parakeet_ctc", "ParakeetForCTC"),
|
|
1585
1621
|
("sew", "SEWForCTC"),
|
|
1586
1622
|
("sew-d", "SEWDForCTC"),
|
|
@@ -1674,6 +1710,7 @@ MODEL_FOR_BACKBONE_MAPPING_NAMES = OrderedDict(
|
|
|
1674
1710
|
("hgnet_v2", "HGNetV2Backbone"),
|
|
1675
1711
|
("hiera", "HieraBackbone"),
|
|
1676
1712
|
("maskformer-swin", "MaskFormerSwinBackbone"),
|
|
1713
|
+
("pixio", "PixioBackbone"),
|
|
1677
1714
|
("pvt_v2", "PvtV2Backbone"),
|
|
1678
1715
|
("resnet", "ResNetBackbone"),
|
|
1679
1716
|
("rt_detr_resnet", "RTDetrResNetBackbone"),
|
|
@@ -68,6 +68,7 @@ PROCESSOR_MAPPING_NAMES = OrderedDict(
|
|
|
68
68
|
("dia", "DiaProcessor"),
|
|
69
69
|
("edgetam", "Sam2Processor"),
|
|
70
70
|
("emu3", "Emu3Processor"),
|
|
71
|
+
("ernie4_5_vl_moe", "Ernie4_5_VL_MoeProcessor"),
|
|
71
72
|
("evolla", "EvollaProcessor"),
|
|
72
73
|
("flava", "FlavaProcessor"),
|
|
73
74
|
("florence2", "Florence2Processor"),
|
|
@@ -78,6 +79,7 @@ PROCESSOR_MAPPING_NAMES = OrderedDict(
|
|
|
78
79
|
("glm46v", "Glm46VProcessor"),
|
|
79
80
|
("glm4v", "Glm4vProcessor"),
|
|
80
81
|
("glm4v_moe", "Glm4vProcessor"),
|
|
82
|
+
("glmasr", "GlmAsrProcessor"),
|
|
81
83
|
("got_ocr2", "GotOcr2Processor"),
|
|
82
84
|
("granite_speech", "GraniteSpeechProcessor"),
|
|
83
85
|
("grounding-dino", "GroundingDinoProcessor"),
|
|
@@ -93,6 +95,8 @@ PROCESSOR_MAPPING_NAMES = OrderedDict(
|
|
|
93
95
|
("kosmos-2", "Kosmos2Processor"),
|
|
94
96
|
("kosmos-2.5", "Kosmos2_5Processor"),
|
|
95
97
|
("kyutai_speech_to_text", "KyutaiSpeechToTextProcessor"),
|
|
98
|
+
("lasr_ctc", "LasrProcessor"),
|
|
99
|
+
("lasr_encoder", "LasrProcessor"),
|
|
96
100
|
("layoutlmv2", "LayoutLMv2Processor"),
|
|
97
101
|
("layoutlmv3", "LayoutLMv3Processor"),
|
|
98
102
|
("layoutxlm", "LayoutXLMProcessor"),
|
|
@@ -114,6 +118,7 @@ PROCESSOR_MAPPING_NAMES = OrderedDict(
|
|
|
114
118
|
("ovis2", "Ovis2Processor"),
|
|
115
119
|
("owlv2", "Owlv2Processor"),
|
|
116
120
|
("owlvit", "OwlViTProcessor"),
|
|
121
|
+
("paddleocr_vl", "PaddleOCRVLProcessor"),
|
|
117
122
|
("paligemma", "PaliGemmaProcessor"),
|
|
118
123
|
("perception_lm", "PerceptionLMProcessor"),
|
|
119
124
|
("phi4_multimodal", "Phi4MultimodalProcessor"),
|
|
@@ -317,7 +322,6 @@ class AutoProcessor:
|
|
|
317
322
|
processor_class = config_dict.get("processor_class", None)
|
|
318
323
|
if "AutoProcessor" in config_dict.get("auto_map", {}):
|
|
319
324
|
processor_auto_map = config_dict["auto_map"]["AutoProcessor"]
|
|
320
|
-
|
|
321
325
|
# Saved as feature extractor
|
|
322
326
|
if preprocessor_config_file is None:
|
|
323
327
|
preprocessor_config_file = cached_file(
|
|
@@ -345,16 +349,24 @@ class AutoProcessor:
|
|
|
345
349
|
processor_auto_map = config_dict["auto_map"]["AutoProcessor"]
|
|
346
350
|
|
|
347
351
|
if processor_class is None:
|
|
348
|
-
#
|
|
349
|
-
|
|
350
|
-
|
|
351
|
-
|
|
352
|
-
|
|
352
|
+
# Last resort: try loading the model config to get processor_class.
|
|
353
|
+
# This handles cases where processor info is only in config.json (not in any
|
|
354
|
+
# preprocessor/tokenizer config files). AutoConfig.from_pretrained may raise
|
|
355
|
+
# ValueError if the model_type is unrecognized or the config is invalid -
|
|
356
|
+
# we catch and ignore this to allow fallback to AutoTokenizer/AutoImageProcessor.
|
|
357
|
+
try:
|
|
358
|
+
if not isinstance(config, PreTrainedConfig):
|
|
359
|
+
config = AutoConfig.from_pretrained(
|
|
360
|
+
pretrained_model_name_or_path, trust_remote_code=trust_remote_code, **kwargs
|
|
361
|
+
)
|
|
353
362
|
|
|
354
|
-
|
|
355
|
-
|
|
356
|
-
|
|
357
|
-
|
|
363
|
+
processor_class = getattr(config, "processor_class", None)
|
|
364
|
+
if hasattr(config, "auto_map") and "AutoProcessor" in config.auto_map:
|
|
365
|
+
processor_auto_map = config.auto_map["AutoProcessor"]
|
|
366
|
+
except ValueError:
|
|
367
|
+
# Config loading failed (unrecognized model_type, invalid config, etc.)
|
|
368
|
+
# Continue to fallback logic below (AutoTokenizer, AutoImageProcessor, etc.)
|
|
369
|
+
pass
|
|
358
370
|
|
|
359
371
|
if processor_class is not None:
|
|
360
372
|
processor_class = processor_class_from_name(processor_class)
|