;ihPSSKrSSKrSSKJr SSKrSSKJr "SS\R5rSSS\R\RS S SS \RS S \RSS.r SSSSSS.SSSSSS.S.r \R"\5rSrS\S\4SjrSSjrS\S\4SjrSrg)N)Path)CustomArgumentParserc\rSrSrSrSrSrg)ConversionStatusrN)__name__ __module__ __qualname____firstlineno__NOT_YET_IMPLEMENTEDREMOVED__static_attributes__r l/home/dmtnaga/Documents/work/airagagent/rag_env/lib/python3.13/site-packages/accelerate/commands/to_fsdp2.pyrrs Grr fsdp_versionfsdp_reshard_after_forwardfsdp_auto_wrap_policyfsdp_cpu_ram_efficient_loadingfsdp_offload_paramsfsdp_state_dict_type"fsdp_transformer_layer_cls_to_wrapfsdp_min_num_paramsfsdp_activation_checkpointing)rrrfsdp_backward_prefetchfsdp_forward_prefetchrrfsdp_sharding_strategyrfsdp_sync_module_statesrrfsdp_use_orig_paramsrTF) FULL_SHARD SHARD_GRAD_OP HYBRID_SHARDHYBRID_SHARD_ZERO2NO_SHARD)rrc~[UR5R5(d[SURS35eUR(dUR c [ S5eUR(dB[UR 5R5(a[SUR S35egg)Nz Config file z not foundz9If --overwrite is not set, --output_file must be providedz Output file z* already exists and --overwrite is not set)r config_fileexistsFileNotFoundError overwrite output_file ValueErrorFileExistsError)argss r_validate_to_fsdp2_argsr/Gs   ! ( ( * *,t/?/?.@ KLL >>d..6TUU >>d4#3#34;;== T-=-=,>>hijj>>rconfigreturncURS05nU(d[RS5 U$0nURSS5S:Xa,[RS5 [RS5 U$UR 5Hup4[ RUS5n[ U[5(dUcUnXBU'M;U[R:Xa[RSUS 35 MjU[R:Xa[RSUS 35 MUc[RSUS 35 XBU'MU[;a[URXD5nXB[ U'M SUS'X S'U$) N fsdp_configz?No FSDP config found in the config file, skipping conversion...rz6Config already specifies FSDP2, skipping conversion...zaIf the config doesn't use new argument names, change `fsdp_version` to `1` and rerun the command.z Argument z0 has been removed in FSDP2, skipping this key...z6 is not yet implemented in FSDP2, skipping this key...z- is not being converted, skipping this key...) getloggerinfowarningitemsARGUMENT_KEY_MAPPING isinstancerrrARGUMENT_VALUE_MAPPING)r0r3new_fsdp_configkeyvalueconversion_statuss rconvert_config_to_fsdp2rBRsj**]B/K  UV O~q)Q.OP o  !'') 044S$? ')9 : :>O>W # 16- .   0 8 8 8 NNYse+[\ ]   0 D D D NNYse+ab c   $ NNYse+XY Z#(C ,,.s377E9>05 6+*.'(ON#+= MrcSnUbURSUS9nO [US9nURS[SSS9 URSS S S S 9 URS [SSS9 UbUR [ S9 U$)Nz0Convert an Accelerate config from FSDP1 to FSDP2zto-fsdp2) descriptionz --config_filez#The config file to convert to FSDP2T)typehelprequiredz --overwrite store_truez&Overwrite the config file if it existsF)actionrFdefaultz --output_filezThe path to the output file to write the converted config to. If not provided, the input file will be overwritten (if --overwrite is set))rErFrJ)func) add_parserr add_argumentstr set_defaultsto_fsdp2_command) subparsersrDparsers rto_fsdp2_command_parserrS~sDK&&z{&K%+> c8]hlm  5    Y  !12 Mrr'c[U5n[R"U5nSSS5 W(d [S5eU$!,(df  N"=f)NzConfig file is empty)openyaml safe_loadr,)r'fr0s r load_configrYs< k a"  /00 M  s ? A cD[U5 [UR5nUR(aURcURUl[ U5n[ URS5n[R"X#5 SSS5 g!,(df  g=f)Nw) r/rYr'r*r+rBrUrVdump)r.r0 new_configrXs rrPrPsnD! )) *F ~~$**2++(0J d $ *  % $ $s 1B B)N)enumloggingpathlibrrVaccelerate.commands.utilsrEnumrrrr;r= getLoggerr r7r/dictrBrSrNrYrPr rrres"  :tyy#">5.66-AA&F0:2/77*N0,44%D#,# # #"   8 $k)D)T)X6ST !r