Ë
    GêñiüŠ  ã                  ó  — d dl mZ d dlmZ d dlmZ ddlmZmZm	Z	m
Z
mZmZmZmZmZ erddlmZ ddlmZ i 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“dd“dd“dd“dd“i 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$“d*d+“d,d+“d-d$“d.d$“¥i d/d0“d1d2“d3d$“d4d5“d6d5“d7d5“d8d5“d9d5“d:d5“d;d5“d<d5“d=d5“d>d?“d@d?“dAd?“dBd?“dCd?“¥ZdD„ ZdEadF„ Z	 dJ	 	 	 	 	 	 	 dKdG„ZdLdH„Z	 	 	 dM	 	 	 	 	 	 	 	 	 dNdI„ZyE)Oé    )Úannotations)Údeepcopy)ÚTYPE_CHECKINGé   )	ÚChunkÚConcatenateÚ"ErnieFuseAndSplitTextVisionExpertsÚMergeModulelistÚPrefixChangeÚ	TransposeÚWeightConverterÚWeightRenamingÚWeightTransform©ÚPreTrainedModel)ÚHfQuantizerÚminimaxÚmixtralÚ
minimax_m2ÚafmoeÚ	qwen2_moeÚdeepseek_v2Údeepseek_v3Údots1Úernie4_5_moeÚglm4_moeÚglm4_moe_liteÚglm_moe_dsaÚ	glm4v_moeÚlongcat_flashÚ
solar_openÚ	qwen3_moeÚqwen3_omni_moeÚqwen3_omni_moe_thinkerÚ
qwen3_nextÚhunyuan_v1_moeÚ	flex_olmoÚolmoeÚ
exaone_moeÚ
rt_detr_v2Úrt_detrÚpp_doclayout_v2Úpp_doclayout_v3Ú	paligemmaÚllavaÚ
aya_visionÚgot_ocr2Úshieldgemma2Úgemma3ÚinternvlÚllava_next_videoÚ
llava_nextÚllava_onevisionÚvipllavaÚmistral3Ú
qwen2_5_vlÚqwen2_vlÚsam3_tracker_videoÚsam3_trackerÚpp_chart2tableÚaltclip_vision_modelÚclip_vision_modelÚchinese_clip_vision_modelÚclipseg_vision_modelÚmetaclip_2_vision_modelÚmlcd_visionÚmlcdÚsiglip_vision_modelÚsiglip2_vision_modelÚxclip_vision_modelÚclipseg_text_modelÚclip_text_modelÚmetaclip_2_text_modelÚsiglip_text_modelÚsiglip2_text_modelÚxclip_text_modelc            %     ó0  — i dt        dd¬«      g“dt        dd¬«      t        dd	¬«      t        d
d¬«      t        dd¬«      g“dt        dd¬«      t        dd	¬«      t        d
d¬«      t        dd¬«      t        dd¬«      g“dt        d¬«      g“dt        d¬«      g“dt        dd¬«      t        dd	¬«      t        dd¬«      t        dd¬«      t        dd¬«      g“dt        dd¬«      t        dd	¬«      t        dd¬«      g“dt        dd¬«      t        dd	¬«      t        dd ¬«      t        dd¬«      g“d!t        d"d#¬«      t        d$d	¬«      t        d%d&¬«      g“d't        d(d)¬«      t        d*d+¬«      t        d,d¬«      g“d-t        d.d¬«      t        d*d+¬«      g“d/t        d0d1¬2«      t        d3d4¬«      g“d5t        d6¬7«      g“d8t        d9d:¬«      t        d;d<¬«      t        d=d>¬«      t        d?d@¬«      t        dAdB¬«      t        dCdD¬«      t        dEdF¬«      t        dGdH¬«      g“dIt        dJdK«      g“dLt        dMdN«      g“dOt        dPdQ«      g“i dRt        dSdT«      t        dUdV«      g“dWt        dXdY«      t        dZd[«      g“d\t        d]d0¬2«      g“d^t        d_d`¬«      t        dadb¬«      t        dcdd¬«      g“det        dfdg«      t        dhdi«      t        djdk«      g“dlt        dmdn«      t        dodpgdqt        dr¬s«      t	        dt¬s«      g¬u«      t        dvgdwt        dr¬s«      g¬u«      g“dxt        dydzgd{t        dr¬s«      t	        dt¬s«      g¬u«      t        d|d}t        dr¬s«      g¬u«      g“d~t        d{d{t        dtdd€¬�«      g¬u«      t        d}d}t        dtdd€¬�«      g¬u«      g“d‚t        dmdn«      t        dƒd„«      t        dodpgdqt        dr¬s«      t	        dt¬s«      g¬u«      t        dvdwt        dr¬s«      g¬u«      g“d…t        d†d‡gdˆt        dr¬s«      t	        dt¬s«      g¬u«      t        d‰dŠt        dr¬s«      g¬u«      g“d‹t        ddŒ«      t        d�dŽ«      t        d�d�«      t        d‘d’«      t        d“d”«      t        d•d–«      t        d—d˜«      t        d™dš«      t        d›dœ«      t        d�dž«      t        dŸd t        drdt¬¡«      g¬u«      t        d¢d£t        drdt¬¡«      g¬u«      t        d¤gd¥d¦gt        dr¬s«      g¬u«      t        d§gd¨d©gt        drdt¬ª«      g¬u«      t        d«d¬gd­d®gt        drdt¬ª«      g¬u«      g“d¯t        d°d±«      t        d²d³«      t        d´dµ«      t        d¶d·«      t        d¸d¹«      t        dºd»«      t        d¼d½«      t        d¾d¿«      t        dÀdÁ«      t        dÂdÃ«      t        dÄdÅ«      t        dÆdÇ«      t        dÈdÉ«      t        dÊdË«      t        dÌdÍ«      t        dÎdÏ«      t        dÐdÑ«      t        dÒdÓ«      t        dÔdÕ«      t        dÖd×«      g“dØt        d²d³«      t        d´dµ«      t        d¶d·«      t        dÙdÚ«      g“dÛg t        d°d±«      ‘t        dÜdÝ«      ‘t        dÞdß«      ‘t        d´dµ«      ‘t        d¶d·«      ‘t        dàdá«      ‘t        dâdã«      ‘t        dädå«      ‘t        dædç«      ‘t        dèdé«      ‘t        dêdë«      ‘t        dìdí«      ‘t        dîdï«      ‘t        dðdñ«      ‘t        dòdó«      ‘t        dôdõ«      ‘t        d¸d¹«      ‘t        dºd»«      ‘t        d¼d½«      ‘t        d¾d¿«      ‘t        dÀdÁ«      ‘t        dÂdÃ«      ‘t        dÄdÅ«      ‘t        dÆdÇ«      ‘t        dÈdÉ«      ‘t        dÊdË«      ‘t        dÌdÍ«      ‘t        dÎdÏ«      ‘t        dÐdÑ«      ‘t        dÒdÓ«      ‘t        dÔdÕ«      ‘t        dÖd×«      ‘“döt        d°d±«      t        dÜdÝ«      t        d´dµ«      t        d¶d·«      g“d÷t        d²d³«      t        d´dø«      t        d¶dù«      t        dÙdÚ«      g“dút        dûdü«      t        dýdþ«      t        dÿg�d t        dr¬s«      g¬u«      t        �dg�dt        dr¬s«      g¬u«      g“¥t        �d�dgdˆt        dr¬s«      t	        dt¬s«      g¬u«      t        �ddŠt        dr¬s«      g¬u«      gt        �d�d«      t        �d�d	«      t        �d
dÝ«      t        �d�d«      t        �d�d«      t        dV�d«      t        �dd[«      t        �d�d«      t        �dgg �d¢t        dr¬s«      g¬u«      g	t        �d�d	¬«      t        �d�d¬«      t        �dg �d¢t        dr¬s«      g¬u«      t        �ddÝ¬«      t        �dd[¬«      t        �d�d¬«      gt        �d�d«      t        �d�d«      t        �d�d«      t        �d�d«      t        �d�d «      t        �d!�d"«      t        �d#�d$«      t        �d%�d&«      t        �d'�d(«      t        �d)�d*«      t        �d+�d,«      t        �d-�d.«      t        �d/�d0«      t        �d1�d2«      t        �d3�d4«      t        �d5�d6«      t        �d7�d8«      t        �d9�d:«      t        �d;�d<«      t        �d=�d>«      t        �d?�d@«      t        �dA�dB«      t        �dC�dD«      t        �dE�dF«      t        �dG�dH«      t        �dI�dJ«      t        �dK�dL«      t        �dM�dN«      t        �dO�dP«      t        �dQ�dR«      gt        �dS�dT«      t        �dU�dV«      t        �dW�dX«      t        �dY�dZ«      t        �d[�d\«      t        �d]gg �d^¢t        dr¬s«      g¬u«      t        �d_gg �d`¢t        dr¬s«      g¬u«      t        �da�db«      t        �dc�dd«      t        �de�df«      t        �dg�dh«      t        �di�dj«      t        �dk�dl«      t        �dm�dn«      t        �do�dp«      t        �dq�dr«      t        �ds�dt«      gt        �du�dv¬«      t        �dw�dx¬«      g�dyœ¥} | dIxx   | �dz   j                  «       z  cc<   | �dzxx   t        �d{�d|¬«      t        �d}�d~¬«      gz  cc<   | dx   j                  «       | �d<   | �dxx   t        d¤�d€«      gz  cc<   | dl   j                  «       | �d�<   | �d�xx   t        �d‚�dƒ«      gz  cc<   | dx   j                  «       | �d„<   | �d„xx   t        �d…�d†«      gz  cc<   | dx   j                  «       | �d‡<   | �d‡xx   t        �dˆd¢¬«      t        �d‰�d…¬«      t        �dŠ�d‹¬«      gz  cc<   | d\   j                  «       | �dŒ<   | �dŒxx   | dx   j                  «       z  cc<   | dx   j                  «       | �d�<   | �d�xx   t        �dŽ�d†«      t        �d��d‹«      gz  cc<   t        j                  «       D ]   \  }}|| v rŒ| |   j                  «       | |<   Œ" | S (�  NÚaltclipzlayer\.zlayers.©Úsource_patternsÚtarget_patternsr/   z^language_model.modelzmodel.language_modelz^language_model.lm_headÚlm_headz^vision_towerzmodel.vision_towerz^multi_modal_projectorzmodel.multi_modal_projectorr6   z^image_newlinezmodel.image_newliner@   Úvision_model)Úprefix_to_removerJ   Ú
text_modelÚvideo_llavaz^image_towerzmodel.image_towerz^video_towerzmodel.video_towerÚfuyuz^vision_embed_tokenszmodel.vision_embed_tokensÚmllamaz^vision_modelzmodel.vision_modelÚemu3z^text_model.modelzmodel.text_modelz^text_model.lm_headz^vqmodelzmodel.vqmodelÚpaddleocr_vlz^mlp_ARzmodel.projectorz^visualzmodel.visualz1^model(?!(\.visual|\.projector|\.language_model))r;   z((?<!_)model(?!\.(language_model|visual))Úcolqwen2ÚmodelÚvlm)rV   Úmodel_prefixz vlm(?!\.(language_model|visual))zvlm.language_modelÚtimm_wrapperÚ
timm_model)Úprefix_to_addÚpi0Ú
state_projzembed_action_time.state_projÚaction_in_projz embed_action_time.action_in_projÚaction_time_mlp_inz$embed_action_time.action_time_mlp_inÚaction_time_mlp_outz%embed_action_time.action_time_mlp_outz&^paligemma_with_expert.paligemma.modelz	model.vlmz)^paligemma_with_expert.gemma_expert.modelz	model.ditz+^paligemma_with_expert.gemma_expert.lm_headzmodel.dit.embed_tokensz(^paligemma_with_expert.paligemma.lm_headz%model.vlm.language_model.embed_tokensÚesmz9encoder.layer.*.attention.self.rotary_embeddings.inv_freqzrotary_embeddings.inv_freqÚdinov3_convnextz(?<!model\.)stageszmodel.stagesÚ
dinov3_vitz(?<!model\.)layer.zmodel.layer.Ú
timesfm2_5Úff0Úfc1Úff1Úfc2Úolmo_hybridÚattention_layer_normÚinput_layernormÚfeedforward_layer_normÚpost_attention_layernormÚqwen3_5_textÚlanguage_modelr=   z'detector_model.vision_encoder.backbone.zvision_encoder.backbone.ztracker_neck.zvision_encoder.neck.ztracker_model.(.+)z\1Út5gemma2_encoderz-(?<!decoder\.)(?<!text_model\.)embed_tokens\.ztext_model.embed_tokens.z5(?<!decoder\.)(?<!text_model\.)(?<!layer)(?<!_)norm\.ztext_model.norm.zA(?<!vision_model.encoder\.)(?<!decoder\.)(?<!text_model\.)layers.ztext_model.layers.r   z.block_sparse_moe.z.mlp.z.experts.*.w1.weightz.experts.*.w3.weightz.experts.gate_up_projr   )Údimr   )rR   rS   Ú
operationsz.experts.*.w2.weightz.experts.down_projr   zmlp.experts.*.gate_proj.weightzmlp.experts.*.up_proj.weightzmlp.experts.gate_up_projzmlp.experts.*.down_proj.weightzmlp.experts.down_projÚqwen3_vl_moeé   T)Ú
check_dimsÚphimoez.gate.weightz.router.weightÚlfm2_moez feed_forward.experts.*.w1.weightz feed_forward.experts.*.w3.weightz!feed_forward.experts.gate_up_projz feed_forward.experts.*.w2.weightzfeed_forward.experts.down_projÚernie4_5_vl_moeÚvision_towerzspatial_linear.0zspatial_linear.fc1zspatial_linear.2zspatial_linear.fc2zspatial_linear.3zspatial_linear.lnztemporal_linear.0ztemporal_linear.fc1ztemporal_linear.2ztemporal_linear.fc2ztemporal_linear.3ztemporal_linear.lnz!(?<!language_model\.)embed_tokenszlanguage_model.embed_tokensz(?<!language_model\.)layerszlanguage_model.layersz(?<!_)(?<!\w)norm\.zlanguage_model.norm.zmlp.gate.weight_1zmlp.vision_moe.gate.weight)Údim0Údim1zmlp.gate.weightzmlp.text_moe.gate.weightz'mlp.moe_statics.e_score_correction_biasz5mlp.text_moe.gate.moe_statics.e_score_correction_biasz7mlp.vision_moe.gate.moe_statics.e_score_correction_biaszexperts.*.down_proj.weightztext_moe.experts.down_projzvision_moe.experts.down_proj)Ú	stack_dimÚ
concat_dimzexperts.*.gate_proj.weightzexperts.*.up_proj.weightztext_moe.experts.gate_up_projzvision_moe.experts.gate_up_projÚdetrzbackbone.conv_encoderÚbackboneÚout_projÚo_projzlayers.(\d+).fc1zlayers.\1.mlp.fc1zlayers.(\d+).fc2zlayers.\1.mlp.fc2zbbox_attention.q_linearzbbox_attention.q_projzbbox_attention.k_linearzbbox_attention.k_projzmask_head.lay1zmask_head.conv1.convzmask_head.gn1zmask_head.conv1.normzmask_head.lay2zmask_head.conv2.convzmask_head.gn2zmask_head.conv2.normzmask_head.adapter1z"mask_head.fpn_stages.0.fpn_adapterzmask_head.lay3z"mask_head.fpn_stages.0.refine.convzmask_head.gn3z"mask_head.fpn_stages.0.refine.normzmask_head.adapter2z"mask_head.fpn_stages.1.fpn_adapterzmask_head.lay4z"mask_head.fpn_stages.1.refine.convzmask_head.gn4z"mask_head.fpn_stages.1.refine.normzmask_head.adapter3z"mask_head.fpn_stages.2.fpn_adapterzmask_head.lay5z"mask_head.fpn_stages.2.refine.convzmask_head.gn5z"mask_head.fpn_stages.2.refine.normzmask_head.out_layzmask_head.output_convr+   zencoder.encoder.(\d+).layerszencoder.aifi.\1.layersÚconditional_detrzself_attn.out_projzself_attn.o_projzencoder_attn.out_projzencoder_attn.o_projz%decoder.layers.(\d+).sa_qcontent_projz*decoder.layers.\1.self_attn.q_content_projz!decoder.layers.(\d+).sa_qpos_projz&decoder.layers.\1.self_attn.q_pos_projz%decoder.layers.(\d+).sa_kcontent_projz*decoder.layers.\1.self_attn.k_content_projz!decoder.layers.(\d+).sa_kpos_projz&decoder.layers.\1.self_attn.k_pos_projzdecoder.layers.(\d+).sa_v_projz"decoder.layers.\1.self_attn.v_projz%decoder.layers.(\d+).ca_qcontent_projz-decoder.layers.\1.encoder_attn.q_content_projz!decoder.layers.(\d+).ca_qpos_projz)decoder.layers.\1.encoder_attn.q_pos_projz%decoder.layers.(\d+).ca_kcontent_projz-decoder.layers.\1.encoder_attn.k_content_projz!decoder.layers.(\d+).ca_kpos_projz)decoder.layers.\1.encoder_attn.k_pos_projzdecoder.layers.(\d+).ca_v_projz%decoder.layers.\1.encoder_attn.v_projz&decoder.layers.(\d+).ca_qpos_sine_projz.decoder.layers.\1.encoder_attn.q_pos_sine_projÚdeformable_detrÚd_finezlayers.\1.mlp.layers.0zlayers.\1.mlp.layers.1Ú
nemotron_hz	backbone.zmodel.zembedding.weightzembeddings.weightzmixer.experts.*.up_proj.weightzmixer.experts.up_projz mixer.experts.*.down_proj.weightzmixer.experts.down_projz'feed_forward.experts.*.gate_proj.weightz%feed_forward.experts.*.up_proj.weightz'feed_forward.experts.*.down_proj.weightzencoder.layersÚlayersÚemb_lnzembeddings.LayerNormzattn.out_projÚfc11Úup_projÚfc12Ú	gate_projÚ	down_projÚnorm1Únorm2Úpost_mlp_layernormz	attn.Wqkv)zself_attn.q_projzself_attn.k_projzself_attn.v_projz
mixer.Wqkvzmixer.out_projzencoder\.pre_encode\.conv\.zencoder.subsampling.layers.zencoder\.pre_encode\.out\.zencoder.subsampling.linear.z7transf_decoder\._embedding\.position_embedding\.pos_enczdecoder.pos_emb.weightz+transf_decoder\._embedding\.token_embeddingzdecoder.embed_tokensz&transf_decoder\._embedding\.layer_normzdecoder.embedding_layernormz*transf_decoder\._decoder\.final_layer_normzdecoder.normz transf_decoder\._decoder\.layerszdecoder.layerszencoder_decoder_proj\.zdecoder.proj.z"encoder\.(.+)\.self_attn\.linear_qzencoder.\1.self_attn.q_projz"encoder\.(.+)\.self_attn\.linear_kzencoder.\1.self_attn.k_projz"encoder\.(.+)\.self_attn\.linear_vzencoder.\1.self_attn.v_projz$encoder\.(.+)\.self_attn\.linear_outzencoder.\1.self_attn.o_projz$encoder\.(.+)\.self_attn\.linear_posz$encoder.\1.self_attn.relative_k_projz$encoder\.(.+)\.self_attn\.pos_bias_uzencoder.\1.self_attn.bias_uz$encoder\.(.+)\.self_attn\.pos_bias_vzencoder.\1.self_attn.bias_vz)decoder\.(.+)\.first_sub_layer\.query_netzdecoder.\1.self_attn.q_projz'decoder\.(.+)\.first_sub_layer\.key_netzdecoder.\1.self_attn.k_projz)decoder\.(.+)\.first_sub_layer\.value_netzdecoder.\1.self_attn.v_projz.decoder\.(.+)\.first_sub_layer\.out_projectionzdecoder.\1.self_attn.o_projz\.second_sub_layer\.query_netz.encoder_attn.q_projz\.second_sub_layer\.key_netz.encoder_attn.k_projz\.second_sub_layer\.value_netz.encoder_attn.v_projz"\.second_sub_layer\.out_projectionz.encoder_attn.o_projz\.third_sub_layer\.dense_inz.mlp.fc1z\.third_sub_layer\.dense_outz.mlp.fc2z\.layer_norm_1\.z.input_layernorm.z\.layer_norm_2\.z.post_attention_layernorm.z\.layer_norm_3\.z.final_layernorm.z\.conv\.batch_normz
.conv.normzlog_softmax\.mlp\.layer0Úproj_outz^vision_model\.zmodel\.vision_tower\.zencoder\.layers\.zlayers\.z\.ls1z
\.lambda_1z\.ls2z
\.lambda_2z(layers\.\d+)\.attn\.proj\.z\1.attention.projection_layer.zattn.qkv.weight)zattention.q_proj.weightzattention.k_proj.weightzattention.v_proj.weightzattn.qkv.bias)zattention.q_proj.biaszattention.k_proj.biaszattention.v_proj.biasz	\.norm1\.z\.layernorm_before\.z	\.norm2\.z\.layernorm_after\.z\.embeddings\.class_embeddingz\.embeddings\.cls_tokenz \.embeddings\.position_embeddingz!\.embeddings\.position_embeddingsz\.embeddings\.patch_embedding\.z,\.embeddings\.patch_embeddings\.projection\.z^language_model\.model\.zmodel\.language_model\.z^language_model\.lm_head\.z	lm_head\.z
^mlp1\.0\.z*model\.multi_modal_projector\.layer_norm\.z
^mlp1\.1\.z(model\.multi_modal_projector\.linear_1\.z
^mlp1\.3\.z(model\.multi_modal_projector\.linear_2\.zLayerNorm.gammazLayerNorm.weightzLayerNorm.betazLayerNorm.bias)ÚjambaÚ
nomic_bertÚjina_embeddings_v3Ú
cohere_asrÚqianfan_ocrÚlegacyrž   z
.weight_g$z".parametrizations.weight.original0z
.weight_v$z".parametrizations.weight.original1r   z,mlp.gate.moe_statics.e_score_correction_biasr   z).block_sparse_moe.e_score_correction_biasz.mlp.e_score_correction_biasr)   zmlp.e_score_correction_biasz mlp.gate.e_score_correction_biasÚhy_v3zmlp\.router\.gate\.weightzmlp\.expert_biaszmlp\.shared_mlp\.zmlp.shared_experts.Úqwen3_5_moe_textÚlagunaz#mlp.experts.e_score_correction_biaszmlp.shared_expert.)r   r   r   r
   r   r   r   r	   ÚcopyÚ_MODEL_TO_CONVERSION_PATTERNÚitems)ÚmappingÚ
model_typeÚbase_patterns      úa/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/transformers/conversion_mapping.pyÚ$_build_checkpoint_conversion_mappingr©   _   s›  € ðXØÜ¨:ÀyÔQð
ðXð 	ÜÐ+CÐUkÔlÜÐ+EÐW`ÔaÜÐ+;ÐMaÔbÜÐ+DÐVsÔtð	
ð	Xð 	ÜÐ+CÐUkÔlÜÐ+EÐW`ÔaÜÐ+;ÐMaÔbÜÐ+DÐVsÔtÜÐ+<ÐNcÔdð
ðXð" 	œl¸NÔKÐLð#Xð$ 	œL¸,ÔGÐHð%Xð& 	ÜÐ+CÐUkÔlÜÐ+EÐW`ÔaÜ¨?ÐL_Ô`Ü¨?ÐL_Ô`ÜÐ+DÐVsÔtð
ð'Xð4 	ÜÐ+CÐUkÔlÜÐ+EÐW`ÔaÜÐ+BÐToÔpð
ð5Xð> 	ÜÐ+CÐUkÔlÜÐ+EÐW`ÔaÜÐ+;ÐMaÔbÜÐ+DÐVsÔtð	
ð?XðJ 	ÜÐ+?ÐQcÔdÜÐ+AÐS\Ô]Ü¨;ÈÔXð
ðKXðT 	Ü¨:ÐGXÔYÜ¨:À~ÔVÜØ TØ 6ôð
ðUXðd 	ÜØ KÐ]sôô ¨:À~ÔVð	
ðeXðp 	Ü¨'ÀÔFÜÐ+NÐ`tÔuð
ðqXðx 	œ°LÔAÐBðyXðz 	Ü¨=ÐJhÔiÜÐ+<ÐNpÔqÜØ 5ÐGmôô Ø 6ÐHoôô Ð+TÐfqÔrÜÐ+WÐitÔuäØ NØ 8ôô Ø KØ Gôð!
ð{Xðd 	ÜØKØ,óð
ðeXðp 	œNÐ+@À/ÓRÐSðqXðr 	”~Ð&;¸_ÓMÐNñsXðt 	Ü˜5 %Ó(Ü˜5 %Ó(ð
ðuXð| 	ÜÐ1Ð3DÓEÜÐ3Ð5OÓPð
ð}XðD 	œÐ7GÐV]Ô^Ð_ðEXðF 	ÜØ JÐ\vôô Ð+;ÐMcÔdäÐ+@ÐRWÔXð
ðGXðV 	ÜÐKÐMgÓhÜÐSÐUgÓhÜÐ_ÐauÓvð
ðWXð` 	ÜÐ/°Ó9Üà*Ø*ð!ð !8ä#Øôô   AÔ&ð	ôô à*ð!ð !5ä#Øôðô
ð
ðaXðV 	Üà4Ø2ð!ð !;Ü+°Ô2´KÀAÔ4FÐGôô Ø @Ø 7Ü+°Ô2Ð3ôð
ðWXðt 	ÜØ :Ø :Ü% a¨°tÔ<Ð=ôô
 Ø 7Ø 7Ü% a¨°tÔ<Ð=ôð
ðuXðL 	ÜÐ/°Ó9Ü˜>Ð+;Ó<Üà*Ø*ð!ð !8Ü+°Ô2´KÀAÔ4FÐGôô Ø 6Ø 4Ü+°Ô2Ð3ôð
ðMXðn 	Üà6Ø6ð!ð !DÜ+°Ô2´KÀAÔ4FÐGôô Ø BØ @Ü+°Ô2Ð3ôð
ðoXðL 	ä˜>¨>Ó:äÐ-Ð/CÓDÜÐ-Ð/CÓDÜÐ-Ð/BÓCÜÐ.Ð0EÓFÜÐ.Ð0EÓFÜÐ.Ð0DÓEäÐ?ÐA^Ó_ÜÐ9Ð;RÓSÜÐ1Ð3IÓJÜØ 3Ø <Ü%¨1°1Ô5Ð6ôô
 Ø 1Ø :Ü%¨1°1Ô5Ð6ôô
 Ø!JÐ KàKØMð!ô " aœL˜>ôô Ø!=Ð >à0Ø2ð!ô ?ÈÐWXÔYÐZôô à0Ø.ð!ð
 4Ø5ð!ô ?ÈÐWXÔYÐZô
ðQ3
ðMXðt 	ÜÐ2°JÓ?Ü˜: xÓ0ÜÐ.Ð0DÓEÜÐ.Ð0DÓEäÐ4Ð6MÓNÜÐ4Ð6MÓNäÐ+Ð-CÓDÜ˜?Ð,BÓCÜÐ+Ð-CÓDÜ˜?Ð,BÓCÜÐ/Ð1UÓVÜÐ+Ð-QÓRÜ˜?Ð,PÓQÜÐ/Ð1UÓVÜÐ+Ð-QÓRÜ˜?Ð,PÓQÜÐ/Ð1UÓVÜÐ+Ð-QÓRÜ˜?Ð,PÓQÜÐ.Ð0GÓHð-
ðuXðd 	Ü˜: xÓ0ÜÐ.Ð0DÓEÜÐ.Ð0DÓEÜÐ:Ð<UÓVð	
ðeXðp 	ð '
ÜÐ2°JÓ?ð'
äÐ/Ð1CÓDð'
ô Ð2Ð4IÓJð'
ô Ð.Ð0DÓEð	'
ô
 Ð.Ð0DÓEð'
ô ÐCÐErÓsð'
ô Ð?ÐAjÓkð'
ô ÐCÐErÓsð'
ô Ð?ÐAjÓkð'
ô Ð<Ð>cÓdð'
ô ÐCÐEuÓvð'
ô Ð?ÐAmÓnð'
ô ÐCÐEuÓvð'
ô  Ð?ÐAmÓnð!'
ô" Ð<Ð>fÓgð#'
ô$ Ø9Ð;lóð%'
ô, Ð4Ð6MÓNð-'
ô. Ð4Ð6MÓNð/'
ô2 Ð+Ð-CÓDð3'
ô4 ˜?Ð,BÓCð5'
ô6 Ð+Ð-CÓDð7'
ô8 ˜?Ð,BÓCð9'
ô: Ð/Ð1UÓVð;'
ô< Ð+Ð-QÓRð='
ô> ˜?Ð,PÓQð?'
ô@ Ð/Ð1UÓVðA'
ôB Ð+Ð-QÓRðC'
ôD ˜?Ð,PÓQðE'
ôF Ð/Ð1UÓVðG'
ôH Ð+Ð-QÓRðI'
ôJ ˜?Ð,PÓQðK'
ôL Ð.Ð0GÓHðM'
ðqXð@
 	ÜÐ2°JÓ?ÜÐ/Ð1CÓDÜÐ.Ð0DÓEÜÐ.Ð0DÓEð	
ðA
XðL
 	Ü˜: xÓ0ÜÐ.Ð0IÓJÜÐ.Ð0IÓJÜÐ:Ð<UÓVð	
ðM
XðX
 	Ü˜;¨Ó1ÜÐ-Ð/BÓCÜà4ð!ñ !8Ü+°Ô2Ð3ôô á6ð!ñ !:Ü+°Ô2Ð3ôð
ñY
Xô~
 á=Ù;ð!ð !DÜ+°Ô2´KÀAÔ4FÐGôô Ù IØ @Ü+°Ô2Ð3ôð
ô  Ñ,©iÓ8Ü™9Ñ&=Ó>ÜÑ+Ð-@ÓAÜ™7¡JÓ/Ü™7¡LÓ1Ü˜6¡<Ó0Ü™8Ð%@ÓAÜÙÙ%óô Ù!, ó!ô
 " aœL˜>ôð
ô. ©8ÑE[Ô\ÜÑ+;ÉXÔVÜÙ ,ó!ô
 " aœL˜>ôô Ñ+;ÐM_Ô`Ü©7ÐD^Ô_Ü©7ÑDXÔYð
ô" Ñ9Ñ;YÓZÜÑ8Ñ:XÓYÜÑUÑWpÓqÜÑIÑKbÓcÜÑDÑFdÓeÜÑHÉ/ÓZÜÑ>Ñ@QÓRÜÑ4Ñ6FÓGÜÑ@ÑB`ÓaÜÑ@ÑB`ÓaÜÑ@ÑB`ÓaÜÑBÑDbÓcÜÑBÑDkÓlÜÑBÑDbÓcÜÑBÑDbÓcÜÑGÑIgÓhÜÑEÑGeÓfÜÑGÑIgÓhÜÑLÑNlÓmÜÑ;Ñ=TÓUÜÑ9Ñ;RÓSÜÑ;Ñ=TÓUÜÑ@ÑBYÓZÜÑ9¹;ÓGÜÑ:¹KÓHÜÑ.Ñ0DÓEÜÑ.Ñ0MÓNÜÑ.Ñ0DÓEÜÑ0±-Ó@ÜÑ6¹ÓDð=
ôB Ñ-Ñ/GÓHÜÑ/±Ó=Ü™8¡]Ó3Ü™8¡]Ó3ÜÑ9Ñ;\Ó]ÜÙ!2Ð 3Û qÜ! aœL˜>ôô
 Ù!0Ð 1Û kÜ! aœL˜>ôô
 ™<Ñ)@ÓAÜ™<Ñ)?Ó@ÜÑ;Ñ=WÓXÜÑ>Ñ@dÓeÜÑ=Ñ?nÓoÜÑ6Ñ8RÓSÜÑ8¹,ÓGÜ™=Ñ*WÓXÜ™=Ñ*UÓVÜ™=Ñ*UÓVð3
ô8 Ù 1Ù 2ôô Ù 0Ù 0ôð	
ó]X€Gðt ˆEƒN�g™hÑ'×,Ñ,Ó.Ñ.ƒNà‰HÓÜÙ(Ù@ô	
ô 	Ù(Ù@ô	
ð	ñ 	Óð & kÑ2×7Ñ7Ó9€G‰NÑØ‰NÓÜÐ@ÑBpÓqð ñ Óð $ IÑ.×3Ñ3Ó5€G‰LÑØ‰LÓÜÑBÑDbÓcðñ Óð $ KÑ0×5Ñ5Ó7€G‰LÑØ‰LÓœnÑ-JÑLnÓoÐpÑpÓð ˜{Ñ+×0Ñ0Ó2€G‰GÑØ‰GÓÜÑ'CÐUfÔgÜÑ':ÑLiÔjÜÑ';ÑMbÔcðñ Óð
 #*¨.Ñ"9×">Ñ">Ó"@€GÑÑØÑÓ 7¨;Ñ#7×#<Ñ#<Ó#>Ñ>Óà Ñ,×1Ñ1Ó3€G‰HÑØ‰HÓÜÑ<Ñ>`ÓaÜÑ+Ñ-BÓCðñ Óô
 %A×$FÑ$FÓ$Hò ;Ñ ˆ
�LØ˜Ñ ØØ% lÑ3×8Ñ8Ó:ˆ�
Òð;ð
 €Nó    Nc                ó^   — t         €
t        «       a t        t         j                  | «      «      S )N)Ú$_checkpoint_conversion_mapping_cacher©   r   Úget)r¦   s    r¨   Ú!get_checkpoint_conversion_mappingr®   n  s(   € ä+Ð3Ü/SÓ/UÐ,ÜÔ8×<Ñ<¸ZÓHÓIÐIrª   c                óh   — t         €
t        «       a | t         v r|st        d| › d�«      ‚|t         | <   y )NzModel type z5 already exists in the checkpoint conversion mapping.)r¬   r©   Ú
ValueError)r¦   r¥   Ú	overwrites      r¨   Ú&register_checkpoint_conversion_mappingr²   u  s>   € ô ,Ð3Ü/SÓ/UÐ,ØÔ9Ñ9Á)Ü˜; z lÐ2gÐhÓiÐiØ7>Ô(¨Ò4rª   c                óÎ   — t        | j                  dd «      }|�Lt        |«      }|�=|dk7  r8t        |«      D ]*  \  }}t	        |t
        «      sŒ|j                  |«      ||<   Œ, |S y )Nr¦   Ú )ÚgetattrÚconfigr®   Ú	enumerateÚ
isinstancer   Úwith_submodel_prefix)r^   r`   r¦   Úmodel_specific_conversionsÚiÚ
conversions         r¨   Ú$extract_weight_conversions_for_modelr½   ‚  sy   € Ü˜Ÿ™ |°TÓ:€JØÐÜ%FÀzÓ%RÐ"à%Ð1°lÀbÒ6HÜ!*Ð+EÓ!Fò b‘��:ä˜j¬,Õ7Ø4>×4SÑ4SÐT`Ó4aÐ.¨qÒ1ðbð *Ð)Ørª   c                ó  — ddl m} g }|�-|j                  «       D ��cg c]  \  }}t        ||¬«      ‘Œ }}}t	        «       }| j                  «       D ]p  \  }	}
t        |
|«      sŒ|
j                  j                  |vsŒ,t        |
|	«      }|€Œ;|j                  |«       |j                  |
j                  j                  «       Œr |r|j                  t        d«      «       |�|j                  |j                  «       «       |S c c}}w )zÕ
    For a given `model`, obtain the weight conversion mapping if any are registered either as a simple renaming
    `_checkpoint_conversion_mapping` class argument, or in the general WeightConverter mapping.
    r   r   rQ   rž   )Úmodeling_utilsr   r¤   r   ÚsetÚnamed_modulesr¸   r¶   r¦   r½   ÚextendÚaddr®   Úget_weight_conversions)r^   Úkey_mappingÚhf_quantizerÚ
add_legacyr   Úweight_conversionsÚkÚvÚseen_model_typesÚnameÚ	submoduleÚconversionss               r¨   Úget_model_conversion_mappingrÏ   �  s  € õ 0ð Ðð ÐØ_j×_pÑ_pÓ_r×sÑW[ÐWXÐZ[œn¸QÐPQÖRÐsÐÑsô “uÐà ×.Ñ.Ó0ò B‰ˆˆiÜ�i Õ1°i×6FÑ6F×6QÑ6QÐYiÒ6iÜ>¸yÈ$ÓOˆKØÑ&Ø"×)Ñ)¨+Ô6Ø ×$Ñ$ Y×%5Ñ%5×%@Ñ%@ÕAðBñ Ø×!Ñ!Ô"CÀHÓ"MÔNð ÐØ×!Ñ! ,×"EÑ"EÓ"GÔHàÐùó) ts   žD)F)r¦   Ústrr¥   z&list[WeightConverter | WeightRenaming]r±   ÚboolÚreturnÚNone)r^   r   r`   rÐ   rÒ   zlist[WeightTransform] | None)NNT)
r^   r   rÅ   zdict[str, str] | NonerÆ   zHfQuantizer | NonerÇ   rÑ   rÒ   zlist[WeightTransform])Ú
__future__r   r¢   r   Útypingr   Úcore_model_loadingr   r   r	   r
   r   r   r   r   r   r¿   r   Ú
quantizersr   r£   r©   r¬   r®   r²   r½   rÏ   © rª   r¨   ú<module>rÙ      sU  ðõ #å Ý  ÷
÷ 
õ 
ñ Ý/Ý'ð6 àˆyð6 ð �)ð6 ð
 ˆ[ð6 ð �;ð6 ð �;ð6 ð ˆ[ð6 ð �Kð6 ð �ð6 ð �[ð6 ð �;ð6 ð �ð6 ð �[ð6 ð �+ð6 ð  �ð!6 ð" �kð#6 ð$ ˜kð%6 ð& �+ñ'6 ð( �kð)6 ð* �ð+6 ð, ˆ[ð-6 ð. �+ð/6 ð0 �)ð16 ð2 �yð36 ð4 �yð56 ð6 �ð76 ð8 �'ð96 ð: �ð;6 ð< �Gð=6 ð> ˆgð?6 ð@ �ðA6 ðB ˜ðC6 ðD �|ðE6 ðF �ðG6 ðH �òI6 ðJ �*ðK6 ðL ˜.ðM6 ðN �gðO6 ðP Ð/ðQ6 ðR  Ð!4ðS6 ðT Ð/ðU6 ðV Ð2ðW6 ðX Ð&ðY6 ðZ Ðð[6 ð\ Ð.ð]6 ð^ Ð/ð_6 ð` Ð-ða6 ðb Ð+ðc6 ðd Ð.ðe6 ðf Ð*ðg6 ðh Ð+ði6 ðj Ð)ñk6 Ð òrIðX (,Ð $òJð ð
?Øð
?à3ð
?ð ð
?ð 
ó	
?óð  *.Ø'+Øð	&Øð&à&ð&ð %ð&ð ð	&ð
 ô&rª   