+ QV-jBRt^RIt^RIt^RIt^RIHtHt^RIHt]P!] 4t !RR]P4t !RR4tRt!R R 4tRR R lltR tR#)zACollection of utils to be used by backbones and their components.N)hf_apilogging)maybe_install_capturing_hooksc]tRt^tRtRtRtR#) BackboneTypetimm transformersN)__name__ __module__ __qualname____firstlineno__TIMM TRANSFORMERS__static_attributes__r l/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/transformers/backbone_utils.pyrrs D!Lrrcaa]tRt^!toRtV3RlRltRt]R4t]PV3RlRl4t]R4t ] PV3R lR l4t V3R lt R t Vt V;t#) BackboneConfigMixinzn A Mixin to support handling the `out_features` and `out_indices` attributes for the backbone configurations. cB<V^8dQhRS[R,RS[R,/#) out_featuresN out_indices)list)format __classdict__s"r __annotate__ BackboneConfigMixin.__annotate__&s'"/"/Tk"/D["/rc *Wn\V\4'd \V4MTVnVP 4VPVPr!Vf7Vf3\ VP4^, .pVPR,.pM[Vf/Ve+VUu.uFq0PPV4NK ppM)Vf&Ve"VUu.uFq@PV,NK ppWuVnVnVP 4R#uupiuupi)a( Sets output indices and features to new values and aligns them with the given `stage_names`. If one of the inputs is not given, find the corresponding `out_features` or `out_indices` for the given `stage_names`. Args: out_features (`list[str]`, *optional*): The names of the features for the backbone to output. Defaults to `config._out_features` if not provided. out_indices (`list[int]` or `tuple[int]`, *optional*): The indices of the features for the backbone to output. Defaults to `config._out_indices` if not provided. N) _out_features isinstancetupler _out_indicesverify_out_features_out_indiceslen stage_namesindex)selfrrlayeridxs&&& r"set_output_features_output_indices6BackboneConfigMixin.set_output_features_output_indices&s *1;K1O1OD-U` ,,.%)$6$68I8Ik  <#7t//0145K ,,R01L  \%=FRSlU++11%8lKSK  !k&==HI[c,,S11[LI1=-D- ,,. TIs #D Dc aSPf \R4hSPEed\SP\34'g"\R\ SP4 24h\ ;QJd)V3RlSP4F 'gK RM RM!V3RlSP44'd&\RSP RSP 24h\SP4\\SP448wd\R SP 24hSPSPUu.uFqSP9gKVNK up;p8wd\R V RSP 24hSPEe\SP\4'g"\R \ SP4 24h\;QJd!.V3R lSP4FNK 5M!V3R lSP44p\ ;QJdV3R lV4F 'gK RM RM!V3R lV44'd&\RSP RSP 24h\V4\\V448wd:RSP 2pYCSP8wdRV R2MR, p\V4hV\\V448wdQ\\VSP4RR7UUu.uFwrVVNK ppp\RV RSP 24hSPeSPe\SP4\SP48wd \R4hSPSPUu.uFpSPV,NK up8wd \R4hR#R#R#uupiuuppiuupi)zO Verify that out_indices and out_features are valid for the given stage_names. Nz2Stage_names must be set for transformers backbonesz out_features must be a list got c3@<"TFqSP9xK R#5iN)r&).0featr(s& r FBackboneConfigMixin.verify_out_features_out_indices..TsOEBackboneConfigMixin.verify_out_features_out_indices..ps efgheir)keyz?out_indices must be in the same order as stage_names, expected zHout_features and out_indices should have the same length if both are setzQout_features and out_indices should correspond to the same stages if both are set) r& ValueErrorr r!rtypeanyr%setr#r"sortedzip)r(r1 sorted_featspositive_indicesmsg_r*sorted_negativesf rr$3BackboneConfigMixin.verify_out_features_out_indicesJs    #QR R    )d004':: #CDI[I[D\C]!^__sOD4 @} r)r r#)r r r r __doc__r+r$propertyrsetterrr`r__classdictcell__ __classcell__rcrs@@rrr!s"/"/H0vd""]] !!\\rrcFa\P!S4V3Rl4pV#)a Wrapper for backbone forwards. Backbones always compute `hidden_states` to build their feature maps, so this forces `output_hidden_states=True` on the wrapped forward and then removes `hidden_states` from the returned object unless the caller explicitly requested them. NOTE: We assume a `can_return_tuple` decorator to be applied before so that we always expect a dict like object to remove the hidden states. c<VPR\VPRR44pRVR&S!V.VO5/VBpV'g>VP4UUu/uFwrVVR8wgKWVbK ppp\ V4!R/VBpV#uuppi)output_hidden_statesFT hidden_statesr )getgetattrconfigitemsr@) r(argskwargsrmrbkvfiltered_output_dataforward_functions &*, rwrapper,filter_output_hidden_states..wrappers%zz*@'$++WmotBuv)-%&!$888#5;\\^#\^TQqOG[DAD^ #\&\9$89F $]s  B%B) functoolswraps)rxrysf rfilter_output_hidden_statesr}s(__%&' Nrc>aa]tRt^toRtRtV3RlV3RlltV3RltV3RlRltV3RlR lt ] R 4t ] PV3R lR l4t ] R 4t ] PV3RlRl4t ] R4t] R4tRtRV3RlRlltV3RltRtVtV;t#) BackboneMixinNTc<V^8dQhRR/#rreturnNr )rrs"rrBackboneMixin.__annotate__sSS4Src <\SV`!V/VBVPRR4pVe\PVnM\P VnVP \P8XdVPVR7R#VP \P 8XdVP4R#\RVP R24h)z Method to initialize the backbone. This method is called by the constructor of the base class after the pretrained model weights have been loaded. timm_backboneN)backbonezbackbone_type z not supported.) r___init__rarr backbone_typer_init_timm_backbone_init_transformers_backboner?)r(rsrtrrcs&*, rrBackboneMixin.__init__s $)&) ?D9  $!-!2!2D !-!:!:D    !2!2 2  $ $m $ <   <#<#< <  , , .~d.@.@-AQR Rrc :<\SV`4\V4R#)am Override `post_init` to always install capturing hooks, as backbone will ALWAYS capture outputs. We need to do it in `post_init`, as modules need to be already instantiated. It avoids some mixups with `torch.compile`, as the first hook installation will need/create a graph break, which can clash with external user call such as `model = torch.compile(model...)`. N)r_ post_initr)r(rcs&rrBackboneMixin.post_inits %d+rc<V^8dQhRR/#rr )rrs"rrrsRRtRrc \VPRR4p\VPRR4pVPPUu.uF qDR,NK upVnVPPUu.uF qDR,NK upVn\ VPP4pVPP4pVeW&8wd\RV R24hVe W0P8wd\RV R 24hVPVPnVPPWe4R#uupiuupi) zZ Initialize the backbone model from timm. The backbone must already be loaded to backbone rNr&modulenum_chsz!Config has `out_features` set to z which doesn't match `out_features` from backbone's feature_info. Please check if your checkpoint has correct out features/indices saved.z Config has `stage_names` set to z which doesn't match `stage_names` from backbone's feature_info. Please check if your checkpoint has correct `stage_names` saved.) rprq feature_infoinfor& num_featuresrr module_namer?r+)r(rout_features_from_configstage_names_from_configstagerrs&& rr!BackboneMixin._init_timm_backbones@ $+4;;#M ")$++}d"K :B9N9N9S9ST9S(OO9ST;C;P;P;U;UV;U%9--;UV800<<= ,,88: # /4L4\34L3MNxx  # .3JN^N^3^23J2KLqq  #'"2"2  66|Q)UVs D>8Ec<V^8dQhRR/#rr )rrs"rrrs!!T!rc~VPPVnVPP4RVnR#r/)rqr&r$rrLs&rr)BackboneMixin._init_transformers_backbones,;;22 335 rc.VPP#r/)rqr rLs&rrBackboneMixin.out_featuress{{(((rc0<V^8dQhRS[S[,/#rOrP)rrs"rrrs00c0rc &WPnR#)rSN)rqrrUs&&rrrs $0  rc.VPP#r/)rqr#rLs&rrBackboneMixin.out_indices s{{'''rcP<V^8dQhRS[S[,S[S[,,/#)rrrY)rrs"rrrs!..uSzDI'=.rc &WPnR#)r\N)rqrr]s&&rrrs #. rc\VP4UUu/uFwrW PV,bK upp#uuppir/) enumerater&r)r(irs& rout_feature_channels"BackboneMixin.out_feature_channelss:=FdFVFVO++6+: O ++?+C Or)NNNN)reenumr{rutilsrrutils.output_capturingr get_loggerr rEnumrrr}rrrr rrrshH "A   H %"499" zzz.|]|]~/#dr