+ jr^RIt^RIt^RIHt^RIt^RIHt!RR]P4tRRRRRRR ]PR ]PR R R R R RRRR]PRRRRR]PRR/t R RRRRRRRRRR/RRRRRRRRRRR//t ]P!]4tRtRRltR"RltRR ltR!tR#)#N)Path)CustomArgumentParserc]tRt^t^tRtRtR#)ConversionStatusN)__name__ __module__ __qualname____firstlineno__NOT_YET_IMPLEMENTEDREMOVED__static_attributes__r]/home/luca/screen-checking/.venv/lib/python3.14/site-packages/accelerate/commands/to_fsdp2.pyrrs Grr fsdp_versionfsdp_reshard_after_forwardfsdp_auto_wrap_policyfsdp_backward_prefetchfsdp_forward_prefetchfsdp_cpu_ram_efficient_loadingfsdp_offload_paramsfsdp_sharding_strategyfsdp_state_dict_typefsdp_sync_module_states"fsdp_transformer_layer_cls_to_wrapfsdp_min_num_paramsfsdp_use_orig_paramsfsdp_activation_checkpointing FULL_SHARDT SHARD_GRAD_OPF HYBRID_SHARDHYBRID_SHARD_ZERO2NO_SHARDc\VP4P4'g\RVP R24hVP'gVP f \ R4hVP'gE\VP 4P4'd\RVP R24hR#R#)z Config file z not foundNz9If --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>>rc0V^8dQhR\R\/#)configreturn)dict)formats"r __annotate__r5Rs++D+T+rcVPR/4pV'g\PR4V#/pVPR^4^8Xd-\PR4\PR4V#VP 4Fwr4\ PVR4pVfWBV&K%V\ P8Xd\PRV R24KUV\ P8Xd\PRV R24KVf \PRV R 24WBV&KV\9d\V,PWD4pWB\ V,&K ^VR&W R&V#) 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.Nz 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_MAPPINGrrr ARGUMENT_VALUE_MAPPING)r1r7new_fsdp_configkeyvalueconversion_statuss& rconvert_config_to_fsdp2rCRsV**]B/K  UV O~q)Q.OP o  !'') 044S$?  $#(C   0 8 8 8 NNYse+[\ ]   0 D D D NNYse+ab c   $ NNYse+XY Z#(C ,,.s377E9>05 6/*2'(ON#+= MrcRpVeVPRVR7pM \VR7pVPR\RRR7VPRR R R R 7VPR \RRR7VeVP \ R7V#)z0Convert an Accelerate config from FSDP1 to FSDP2Nzto-fsdp2) descriptionz --config_filez#The config file to convert to FSDP2T)typehelprequiredz --overwrite store_truez&Overwrite the config file if it existsF)actionrGdefaultz --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))rFrGrK)func) add_parserr add_argumentstr set_defaultsto_fsdp2_command) subparsersrEparsers& rto_fsdp2_command_parserrTsDK&&z{&K%+> c8]hlm  5    Y  !12 Mrc0V^8dQhR\R\/#)r0r&r2)rOr3)r4s"rr5r5sSTrc\V4;_uu_4p\P!V4pRRR4X'g \R4hV# +'giL%;i)NzConfig file is empty)openyaml safe_loadr+)r&fr1s& r load_configr[s? k  a"  /00 M  s A A c`\V4\VP4pVP'd VPfVPVn\ V4p\ VPR4;_uu_4p\P!W#4RRR4R# +'giR#;i)Nw) r.r[r&r)r*rCrWrXdump)r-r1 new_configrZs& rrQrQsvD! )) *F ~~~$**2++(0J d $ $ *  % $ $ $s ;B B- )N)enumloggingpathlibrrXaccelerate.commands.utilsrEnumrrr r=r> getLoggerr r9r.rCrTr[rQrrrrfs"  :tyyN ">4.66-AA$&F0:2/77(*N0,44#%D#*deE !deE #"   8 $k+\6 !r