RЦi`SSKrSSKrSSKJr SSKJr SSKJrJrJ r /SQr Sr Sr Sr S rg) N)deepcopy)nn) Conv2dSameBatchNormAct2dLinear) extract_layer set_layeradapt_model_from_stringadapt_model_from_filecPURS5nUn[US5(aUSS:wa URn[US5(dUSS:XaUSSnUHGn[X#5(a2UR5(d [ X#5nM5U[ U5nMEUs $ U$)zExtract a layer from a model using dot-separated path. Args: model: PyTorch model. layer: Dot-separated layer path (e.g., 'layer1.0.conv1'). Returns: Extracted module. .modulerN)splithasattrrisdigitgetattrint)modellayerrls Q/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/timm/models/_prune.pyrr s KK E FuhE!H$8 5( # #aH(<ab   6  99;; +AM McURS5nUn[US5(aUSS:wa URnSnUnUHIn[XV5(dMUR5(d [ XV5nOU[ U5nUS- nMK US-nUSUH3nUR5(d [ X65nM%U[ U5nM5 Xn[ X6U5 g)zSet a layer in a model using dot-separated path. Args: model: PyTorch model. layer: Dot-separated layer path. val: New value for the layer. r rrrN)rrrrrrsetattr)rrvalr lst_indexmodule2rs rr r 's KK E FuhE!H$8IG  7  99;;!'-!#a&/ NI NI :I yy{{V'FCF^F  A Fsrc Sn0nURU5nUHYnURS5nUSnUSSSRS5nUSS:wdM;UVs/sHn[U5PM snX6'M[ [UR55Rn [UR55R n XS.n [ U5n UR5GHrup[X 5n[U[R5(d[U[5(a[U[5(a[nO[RnX=S -nUSnUSnSnURS:aUnUnU"SUUURURS LUR UR"UUR$S .U D6n['XU5 M[U[(5(ah[)X=S -S4UR*UR,UR.S S .U D6nUR0UlUR2Ul['XU5 GMh[U[R45(aQ[R4"SX=S -SUR*UR,UR.S S.U D6n['XU5 GM[U[R65(dGMX=S -Sn[7SUUR8URS LS.U D6n['XU5 [;U S5(dGMI[=U SS5U R>:XaUU l UU lGMu U RC5 URC5 U $s snf)zAdapt a model to pruned structure from string specification. Args: parent_module: Original model to adapt. model_string: String containing layer shapes for pruned model. Returns: Adapted model with pruned layer dimensions. z***:rr,)devicedtypez.weightN) in_channels out_channels kernel_sizebiaspaddingdilationgroupsstrideT)epsmomentumaffinetrack_running_stats) num_featuresr.r/r0r1) in_features out_featuresr)r2head_hidden_size)"rrnext parametersr$r%r named_modulesr isinstancerConv2drr,r(r)r*r+r-r rr.r/r0dropact BatchNorm2drr4rrr2r5eval) parent_module model_string separator state_dict lst_shapekkeyshapeir$r%dd new_modulenm old_moduleconvsr&r'gnew_convnew_bnr2new_fcs rr r FsEIJ""9-I  GGCLd!Qr   % 8r>/45u!s1vu5JO -**, - 4 4F ))+ , 2 2E +B-(J++-"=4 j")) , , :z0R0R*j11!yyy=)AA$KQ4LA  1$*  ')&22__D0"**#,,!((  H jX .  N 3 3#y=)!,NN#,,!(($(   F%//FK#FJ jV ,  BNN 3 3^^'I 6q9NN#,,!(($(   F jV ,  BII . .%)m4Q7L('44__D0 F jV ,z>22:'91=AXAXX2>J/*6 '{.~OO U6sM3c[R"[[RR SUS-55n[ XRS5R55$)zAdapt a model to pruned structure from file specification. Args: parent_module: Original model to adapt. model_variant: Name of pruned model variant file. Returns: Adapted model with pruned layer dimensions. _prunedz.txtzutf-8) pkgutilget_data__name__ospathjoinr decodestrip)r@ model_variant adapt_datas rr r sJ!!(BGGLLMTZDZ,[\J "=2C2CG2L2R2R2T UUr)rYrVcopyrtorchr timm.layersrrr__all__rr r r r6rrrds3 :: \6>\~ Vr