RЦiSrSSKrSSKrSSKJrJrJrJrJr SSK r SSK r Sr \R"\5r/SQrS\S\S \4S jrS \\\4S \\\44S jrSS \S\S\\\ R,4S\S \\\44 SjjrSS\ R0R2S \S\S\\\ R,4S\S\S\\S\S \4SjjrSS \\\4S\ R0R2S\S \\\44SjjrS S\ R0R2S \S\\ R8R:S\\S\S \\4 Sjjrg!\ a Sr GN"f=f)!zhModel creation / weight loading / state_dict helpers Hacked together by / Copyright 2020 Ross Wightman N)AnyCallableDictOptionalUnionTF)clean_state_dictload_state_dictload_checkpointremap_state_dictresume_checkpointtextprefixreturncNURU5(aU[U5S$U$)N) startswithlen)r rs S/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/timm/models/_helpers.py_remove_prefixrs' vCKL!! K state_dictcp0nSnUR5Hup4UHn[X55nM XAU'M U$)N)zmodule.z _orig_mod.)itemsr)rcleaned_state_dict to_removekvrs rrrsJI  "Aq$A !1# rcheckpoint_pathuse_emadevice weights_onlycU(Ga([RRU5(Ga[U5R S5(a0[ (dS5e[ RRXS9nO[R"XUS9nSn[U[5(aKU(aURSS5bSnO.U(aURS S5bS nOS U;aS nOS U;aS n[U(aXEOU5n[R!S R#XP55 U$[R%S R#U55 ['5e![a [R"XS9nNf=f)a?Load state dictionary from checkpoint file. Args: checkpoint_path: Path to checkpoint file. use_ema: Whether to use EMA weights if available. device: Device to load checkpoint to. weights_only: Whether to load only weights (torch.load parameter). Returns: State dictionary loaded from checkpoint. z .safetensorsz-`pip install safetensors` to use .safetensors)r  map_locationr!)r$state_dict_emaN model_emarmodelzLoaded {} from checkpoint '{}'No checkpoint found at '{}')ospathisfilestrendswith_has_safetensors safetensorstorch load_fileload TypeError isinstancedictgetr_loggerinfoformaterrorFileNotFoundError)rrr r! checkpointstate_dict_keyrs rr r ,sB"277>>/::   ( ( 8 8## T%T T#$**44_4TJ N"ZZ[gh  j$ ' ':>>*:DAM!1Z^^K>J!,+!-J&!(%Nj&@Xbc  5<<^]^ 3::?KL!!% N"ZZM  Ns>EF?Fr(strictremap filter_fnc@[RRU5SR5S;a.[ US5(aUR U5 g[ S5e[XX7S9nU(a [X5nOU(aU"X5nURXS9n U $)a Load checkpoint into model. Args: model: Model to load checkpoint into. checkpoint_path: Path to checkpoint file. use_ema: Whether to use EMA weights if available. device: Device to load checkpoint to. strict: Whether to strictly enforce state_dict keys match. remap: Whether to remap state dict keys by order. filter_fn: Optional function to filter state dict. weights_only: Whether to load only weights (torch.load parameter). Returns: Incompatible keys from model.load_state_dict(). )z.npzz.npyload_pretrainedz"Model cannot load numpy checkpointN)r r!)r?) r*r+splitextlowerhasattrrDNotImplementedErrorr r ) r(rrr r?r@rAr!rincompatible_keyss rr r Zs2 ww(,2248HH 5+ , ,  ! !/ 2 &&JK K!&dJ %j8 z1 --j-H r allow_reshapec 0n[UR5R5UR55HuupEupgUR5UR5:Xd(SUSURSUSURS3 5eURUR:waKU(aUR UR5nO(SUSURSUSURS3 5eXsU'M U$)a{Remap checkpoint by iterating over state dicts in order (ignoring original keys). This assumes models (and originating state dict) were created with params registered in same order. Args: state_dict: State dict to remap. model: Model whose state dict keys to use. allow_reshape: Whether to allow reshaping tensors to match. Returns: Remapped state dictionary. zTensor size mismatch z: z vs z. Remap failed.zTensor shape mismatch )ziprrnumelshapereshape)rr(rJout_dictkavakbvbs rr r s"H!%"2"2"4":":"N>N>PQ(2xxzRXXZ't+@BrxxjPTUWTXXZ[][c[cZdds)tt' 88rxx ZZ)j 6rd"RXXJd2$bQSQYQYPZZijju R Or optimizer loss_scalerlog_infocSn[RRU5(Ga[R"USSS9n[ U[ 5(GaSU;GaU(a[RS5 [US5nURU5 Ub6SU;a0U(a[RS5 URUS5 UbIURU;a9U(a[RS 5 URXcR5 S U;aGUS nS U;aUS S :aUS - nU(a'[RS RXS 55 U$URU5 U(a$[RSRU55 U$[RSRU55 [5e)aVResume training from checkpoint. Args: model: Model to load checkpoint into. checkpoint_path: Path to checkpoint file. optimizer: Optional optimizer to restore state. loss_scaler: Optional AMP loss scaler to restore state. log_info: Whether to log loading info. Returns: Resume epoch number if available, else None. NcpuFr#rz(Restoring model state from checkpoint...rUz,Restoring optimizer state from checkpoint...z2Restoring AMP loss scaler state from checkpoint...epochversionz!Loaded checkpoint '{}' (epoch {})zLoaded checkpoint '{}'r))r*r+r,r1r3r5r6r8r9rr r>r:r;r<)r(rrUrVrW resume_epochr=rs rr r s&L ww~~o&&ZZeRWX j$ ' 'LJ,F GH)*\*BCJ  ! !* -$ )BLL!OP))*[*AB&;+E+E+SLL!UV++J7Q7Q,RS*$)'2  *z)/Dq/H A%LLL!D!K!KOho]p!qr   ! !* - 5<<_MN 3::?KL!!r)TrYF)TrYTFNF)T)NNT) __doc__loggingr*typingrrrrrr1safetensors.torchr0r/ ImportError getLogger__name__r8__all__r-rrboolr r nnModuler r optim Optimizerintr rrrms# 77    H % mcc c3h DcN  +0" +"+"+"c5<<'(+" +"  #s(^ +"b+0(,"'xx'''c5<<'( '  '  'H%'' 'Z#cNxx #s(^ @6:%) 4"xx4"4"EKK1124"c] 4"  4" c] 4"esD;;EE