shzSSKJr SSKJr SSKJrJrJrJrJ r \ \ \4r S\S\ \\ 44SjrS\S\ S\4S jrg ) )STATE_DICT_TYPE)_version)OBJ_PATH set_elementSTATE_DICT_ITEMtraverse_state_dicttraverse_state_dict_v_2_3 state_dictreturnc^^0m0mS[S[SS4UU4Sjjn[RSL=(a [RS:HnU(a[ X5 TT4$[ X5 TT4$)ao Flatten ``state_dict`` made of nested dicts and lists into a top level dictionary. Use ``unflatten_state_dict`` to revert this process. Returns: A tuple with the flatten state_dict and a mapping from original to new state_dict. N.B. The new keys are derived from the object paths, joined by dot. For example: ``{ 'a': {'b':...}}`` results in the key `a.b`. pathvaluer Nc>SR[[U55nUT;a[SU35eUTU'UTU'g)N.zduplicated flatten key )joinmapstr ValueError)rrnew_fqn flattenedmappingss }/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/torch/distributed/checkpoint/_nested_dict.py flat_copy%flatten_state_dict..flat_copy*sF((3sD>* i 6wi@A A" ' 2_3)rrr_derived_versionr r )r r use_v_2_3rrs @@rflatten_state_dictr s"$I "H!!!T!! !!-T(2K2Ku2T!*8 h  J2 h rmappingcZ0nUR5Hup4[X!UU5 M U$)zaRestore the original nested state_dict according to ``mapping`` and the flattened ``state_dict``.)itemsr)r r!nestedkeyrs runflatten_state_dictr&>s1!F &&( FCL%0) MrN)%torch.distributed.checkpoint.metadatarr _traverserrrr r dictrFLATTEN_MAPPINGtupler r&rrr.sjBsH}%   ?O +, F*9r