|-j@hddlmZddlmZddlmZddlmZddlm Z m Z m Z m Z m Z mZmZmZmZer ddlmZddlmZid d d d d d dd dd dd dd dd dd dd dd dd dd dd dd dd dd idd dd dd dd d d d!d"d#d"d$d"d%d&d'd(d)d(d*d(d+d(d,d(d-d(d.d(d/d(id0d(d1d2d3d2d4d2d5d2d6d2d7d8d9d8d:d8d;d8dd8d?d8d@dAdBdAdCdDidEdDdFdDdGdDdHdDdIdDdJdDdKdDdLdDdMdDdNdDdOdDdPdDdQdRdSdTdUdVdWdVdXdVdVdVdVdVdYZdZZd[ad\Z dsdtdfZdudjZ dvdwdrZd[S)x) annotations) defaultdict)deepcopy) TYPE_CHECKING) Chunk Concatenate"ErnieFuseAndSplitTextVisionExpertsMergeModulelist PrefixChange TransposeWeightConverterWeightRenamingWeightTransformPreTrainedModel) HfQuantizerminimaxmixtral minimax_m2afmoe qwen2_moe deepseek_v2 deepseek_v3dots1 ernie4_5_moeglm4_moe glm4_moe_lite glm_moe_dsa glm4v_moe longcat_flash solar_open qwen3_moeqwen3_omni_moeqwen3_omni_moe_thinker qwen3_nexthunyuan_v1_moe flex_olmoolmoe exaone_moe cohere2_moe rt_detr_v2rt_detrpp_doclayout_v2pp_doclayout_v3sam3_tracker_video sam3_trackeraltclip_vision_modelclip_vision_modelchinese_clip_vision_modelclipseg_vision_modelmetaclip_2_vision_model mlcd_visionmlcdsiglip_vision_modelsiglip2_vision_modelxclip_vision_modelclipseg_text_modelclip_text_modelmetaclip_2_text_modelsiglip_text_modelsiglip2_text_modelxclip_text_model paligemmallava aya_visiongot_ocr2gemma3internvlvipllavamistral3pp_chart2tablellava_next_video llava_nextllava_onevisionPaliGemmaModel LlavaModelAyaVisionModel GotOcr2Model Gemma3Model InternVLModel VipLlavaModel Mistral3ModelPPChart2TableModelLlavaNextModelLlavaNextVideoModelLlavaOnevisionModel FuyuModel MllamaModelMaskFormerDetrDecoder DetrModel"Qwen2_5_VLForConditionalGenerationQwen2VLForConditionalGenerationASTModelViTModel BeitModel DeiTModel) IJepaModel ViTMAEModel ViTMSNModel VivitModelc(<:idtdddgtdgtdgd tdgtd d gd tddtddtddtddtddtddtddgdtddgdtddtddtddtddtddtddgd td!d"td#d$td%d&td'd(td)d*td+d,tddtddtddtddtddtddg d-td.d/td0d1td!d"td#d$td%d&td'd(td2d3td4d5td6d7g d8td9d:td;d<td=d>td?dgd@tdAdBtdCdDtdEdFtdGd:tdHd<tdId>tdJdKtdLdMtddtdNdtdOdtdPdQtdRdSg dTtdUdVgdWtdGd:tdHd<tdId>tdXdYtddtddtddgdZtdd[td+d\gd]td^d gd_gtd`da tdbdc tddde tdfdg tdhdi tdjdk tdldm tdndo tdpdq tdrds tdtdu tdvdw tdxdy tdzd{ td|d} td~d tdd tdd tdd tdd tdd tdd tdd tdd tdd tdd tdd tdd tdd tdd tdd tdd tdd tdd tdd tdd tdd tdd tdd tddgdtdt dgtddtdgdtdd gdtdd tdd tdd tdd gdtdd tdd tdd tdd tdd gdt dìĦgidt dƬĦgdtdd gdtdd tdd tddʬ tdd̬ tdd gdtdd tdd tddϬ gdtdd tdd tddҬ tdd gdtddƬ gdtdd tddج tddڬ gdtddݬ tdd߬ tdd gdtdd߬ tdd gdt ddgdt ddgdt dgdtdd tdd tdd tdd tdd tdd tdd tdd gdtddgdtddgdtddgdtddtd d gid td d tddgdt ddgdtdd tdd tdd gdtddtddtddgdtd d!td"d#gd$tdt dgtd%gd&tdggd'td(d)gd*tdt dgtd+d,tdggd-td*d*t dd.d/0gtd,d,t dd.d/0ggd1td d!td2d3td"d#gd$tdt dgtd%d&tdggd4td5d6gd7tdt dgtd8d9tdggd:tdÐd;td<d=td>d?td@dAtdBdCtdDdEtdFdGtdHdItdJdKtdLdMtdNdOt ddPgtdQdRt ddPgtdSgdTdUgtdgtdVgdWdXgtddYgtdZd[gd\d]gtddYggd^td_d`tdadbgdctdddetdfdgtd_d`tdadbgdhtdfdgtd_d`tdadbtdidjgdktdldmtdndotdpdqtdrdstdtdutdvdwtdxdytdzd{td|d}td~dtddtddtddtddtddtddtddtddtddtddtddtddtddtddtdgdtdgtdgdtdggdtdd gdtdd tddtddtddtddtddtddtddtddtddg dtdddetddtddtd_d`tdadbtddtddtddtddtddætdĐdŦtdƐdǦtdȐdɦtdʐd˦td̐dͦtdΐdϦgtdddetddtd_d`tdadbgtdfdgtd_dЦtdadѦtdidjgtdҐdtdӐdԦtdgdtdgtdgdtdggtdِdgd7tdt dgtdېd9tdggtdܐdݦtdސdߦtddtddtddtd dtddtddtdggdtdgg tdސd߬ tdܐdݬ tdd tdd tdd tdgdtdggtddtddtddtddtddtddtddtddtddtddtddtddtddtddtdd td d td d tddtddtddtddtddtddtddtddtddtd d!td"d#td$d%td&d'gtd(d)td*d+td,d-td.d/td0d1td2d3td4d5td6d7td8d9td:d;td<d=td>d?td@dAtdBdCtdDdEtdFggdGtdgtdHggdItdggtdJdK tdLdM gdN }|dxx|dOz cc<|dOxxtdPdQ tdRdS gz cc<tdTdUtdVdWtdXdYtdZd[td\d]td^d_td`datdbdctdddetdfdgtdhditdjdktdldmtdndotdpdqtdrdsg|dt<|dt|du<|d'|dv<|dvxxtdSdwgz cc<|d tdxdytdzd{gz|d|<|d |d}<|d}xxtddQtddStdd~gz cc<|d|d<|dxxtddgz cc<|d'|d<|dxxtddgz cc<|d'|d<|dxxtddQ tdd tdd gz cc<|d|d<|dxx|d'z cc<|d'|d<|dxxtddtddgz cc<t D]'\}}||vr ||||<(|S(Nhrm_textzmlp.gate_up_proj.weightzmlp.gate_proj.weightzmlp.up_proj.weightr)dim)source_patternstarget_patterns operationszattn.gqkv_proj.weight)zself_attn.gate_proj.weightself_attn.q_proj.weightself_attn.k_proj.weightself_attn.v_proj.weightz\.attn\.o_proj\.z.self_attn.o_proj.rkrlrazencoder\.layer\.zlayers.zattention.queryq_projz attention.keyk_projzattention.valuev_projzattention.output.densezattention.o_projzintermediate.densezmlp.fc1z output.densezmlp.fc2ViTMSNForImageClassificationz ^encoder\.z vit.encoder.ViTMAEForPreTraining BeitBackbonez ^fpn1\.0\.zfpn.fpn1.conv_transpose1.z ^fpn1\.1\.zfpn.fpn1.normalization.z ^fpn1\.3\.zfpn.fpn1.conv_transpose2.z ^fpn2\.0\.z fpn.fpn2.z^encoder\.layer\.zbeit.encoder.layer.z ^embeddings\.zbeit.embeddings.BeitForSemanticSegmentationz (?     7!!! "aLLL>    +>Pd e e e! f &  . : : ,h 7 7 ?H 5 5 ,h 7 7 35G H H / ; ; >9 5 5 'f 8 ' =. 9 9) 9f >  ,h 7 7 ?H 5 5 ,h 7 7 35G H H / ; ; >9 5 5 ! ?f N  =*E F F =*C D D =*E F F =+ 6 6 /1F G G +-? @ @ ,h 7 7 ?H 5 5 ,h 7 7 35G H H / ; ; >9 5 5  Of j & >@P Q Q FH\ ] ] =*E F F =*C D D =*E F F =+ 6 6 46S T T CL   ;H  ( kf L  68J K K 46H I I 68J K K -/A B B  Mf X  =?] ^ ^ 24H I I 68O P P 13E F F /1C D D 13E F F .0a b b 68a b b 35G H H < 3 3 < 3 3 >+= > > >+< = = Yf v +^9 5 5  yf P  =/ : : +-? @ @ Qf X  :y Q Q Q Yf ^ v 6 +=Od e e e7v 8 + + =   Cv J  = F   Kv R +I[r s s sSv T  @Rk   Uv Z  ASm   [v ` +HZp q q qav b +J\t u u ucv d  @Rk   ev j  9 7   kv r  8 1   sv D  H <   Ev L  Y R   Mv T  W W   Uv \  S J   ]v d  G J   ev l  P J   mv t  N O   uv F  K C   Gv N  K C   Ov V  J B   Wv ^  L D   _v f  K C   gv n  K C   ov v  D @   wv ~  D @   v F  C ?   Gv N  D @   Ov V  D @   Wv f  F @   gv r  C M   sv z  L J   {v B  L J   Cv J  L H   Kv R --!!<+222KA4F4F4FG    Sv b  9 8+2223   cv _f L  +CUe f f f M f R  +EW` a a a +=Oe f f f +;Ma b b b +DVs t t t  S f ^  +EW` a a a +=Oe f f f +;Ma b b b +DVs t t t +K] ^ ^ ^ ; X X X a f j  :GX Y Y Y :~ V V V  T 6    k f z * :~ V V V  FXn   , { f F \7OOOPG f H G'RRRSI f J LAAABK f L  =Jh i i i ++; < < **!!8+222KA4F4F4FG      6 4+2223    _f @  66!!D+222KA4F4F4FG      B @+2223    Af ^  >> : : -/C D D -/C D D -/B C C .0E F F .0E F F .0D E E ?A^ _ _ 9;R S S 13I J J  3 <%115556     1 :%115556    !J KKM!"aLLL>     != >02!?WXYYYZ     0.! 45!?WXYYYZ   Q3 _f T  .0D E E .0D E E Uf \  2J ? ? :x 0 0 .0D E E .0D E E  ]f h  :x 0 0 .0D E E .0D E E : > 8:c d d 8:d e e :S T T 46U V V 02E F F  9 n n n!aLLL>     ; t t t!aLLL>   S. uf R # 3 I I I% Sf X ( +CU] ^ ^ ^ PRq r r PRq r r NPl m m DF\ ] ] FHc d d GId e e FH` a a DF\ ] ] 57K L L * Yf p  2J ? ? /1C D D 24I J J .0D E E .0D E E CEr s s ?Aj k k CEr s s ?Aj k k <>c d d CEu v v ?Am n n CEu v v ?Am n n <>f g g 9;l  %! qf f ^ 2J ? ? /1C D D .0D E E .0D E E   :x 0 0 .0I J J .0I J J : > +-@ A A 7J / / 7L 1 1 6< 0 0 8%@ A A %   !, !!! "aLLL>    . 8E[ \ \ \ +;X V V V +;M_ ` ` ` 7D^ _ _ _ 7DX Y Y Y  ,!!! "aLLL>     " 9;Y Z Z 8:X Y Y UWp q q IKb c c DFd e e H/ Z Z >@Q R R 46F G G @B` a a @B` a a @B` a a BDb c c BDk l l BDb c c BDb c c GIg h h EGe f f GIg h h LNl m m ;=T U U 9;R S S ;=T U U @BY Z Z 9; G G :K H H .0D E E .0M N N .0D E E 0- @ @ 6 D D= B -/G H H / = = 8] 3 3 8] 3 3 9;\ ] ] <)@ A A <)? @ @ ;=W X X >@d e e =?n o o 68R S S 8, G G =*W X X =*U V V =*U V V !2 3 q q q!aLLL>    !0 1 k k k!aLLL>   + 8  1 2     0 0     yf f f GP ENNNgh',,...NNN H(@    (@     02IJJ02IJJ')?@@(>??')?@@(>??+-QRR')MNN(LMM+-QRR')MNN(LMM+-QRR')MNN(LMM*,CDD!&G !"$188M0N0S0S0U0UG ,-%k27799GN N@Bpqq #:.335579OPP U A  9GKz*//11GG Gw 233w 122}h// $I.3355GL LBDbcc$K05577GL Ln-JLnoopp{+0022GG G'CUfggg':Lijjj';Mbccc #*."9">">"@"@G  7;#7#<#<#>#>> ,1133GH H<>`aa+-BCC %A$F$F$H$H;; L  %l388::  NNczttatt|S)N)$_checkpoint_conversion_mapping_cacherrget)rs r!get_checkpoint_conversion_mappingrs1+3/S/U/U, 8<z0get_model_conversion_mapping..As-sssW[WXZ[nQPQRRRsssrc3TK|]"}|dkp|dzV#dS.N startswithr seen module_names r z/get_model_conversion_mapping..Ts>iiDtrz?[33D3J??iiiiiirc3TK|]"}|dkp|dzV#dSrrrs rrz/get_model_conversion_mapping.._sO""EI @k44TCZ@@""""""rrr)modeling_utilsrrrlist named_modules isinstancerrrranyr scope_prefixextendappendupdate_weight_conversions)rrrrrweight_conversionsseen_identifiers submodulerrrfound_via_class is_root_model transformrs @rget_model_conversion_mappingr()s-"0/////ss_j_p_p_r_rsss5@4E4E"'"5"5"7"7+=+= Y)_55  )__- %0  iiiiL\]gLhiii i i   8 CC %T1 L c""""M]^hMi""" %? KK   #r)  5( 5 5 )4 &&!!+...$++K888  =: = Z ( / / < < <O!!"CH"M"MNNN )CCDVWW r)F)rrrrrrrr)rrrr)NNT) rrrrrrrrrr) __future__r collectionsrrrtypingrcore_model_loadingrr r r r r rrrrr quantizersrrrrrrrr(rrr/s#"""""######                       (//////''''''M  yM )M   [ M  ; M ;M  [M KM  M [M ;M M [M +M  !M "k#M $k%M &+'M M (k)M *+M , [-M .+/M 0;1M 2)3M 4y5M 6y7M 8.9M :/;M < !4=M >/?M @2AM B&CM D EM F.GM H/IM M M J-KM L+MM N.OM P*QM R+SM T)UM VWM X'YM Z[M \ g]M ^_M `aM bcM dgeM f gM h|iM llmM M M nloM pLqM r