9ilddlmZddlZddlmZmZddlmZmZddl m Z dZ d ed fd e d e fd ZdZdZy))PathN)is_npu_availableis_xpu_available) ClusterConfigdefault_json_config_file)SubcommandHelpFormatterzFCreate a default config file for Accelerate with only a few flags set.noF save_locationuse_xpuct|}|jjdd|jrt d|dy|j }|dvrt d|d|d }tjjr9tjj}||d <d|d <|d kDrd |d<nd|d<ntr;|r9tjj}||d <d|d <|d kDrd|d<nZd|d<nTtr9tjj}||d <d|d <|d kDrd|d<nd|d<nd}d|d <d |d <d|d<d|d<tdi|}|j!||S)aA Creates and saves a basic cluster config to be used on a local machine with potentially multiple GPUs. Will also set CPU if it is a CPU-only machine. Args: mixed_precision (`str`, *optional*, defaults to "no"): Mixed Precision to use. Should be one of "no", "fp16", or "bf16" save_location (`str`, *optional*, defaults to `default_json_config_file`): Optional custom save location. Should be passed to `--config_file` when using `accelerate launch`. Default location is inside the huggingface cache folder (`~/.cache/huggingface`) but can be overriden by setting the `HF_HOME` environmental variable, followed by `accelerate/default_config.yaml`. use_xpu (`bool`, *optional*, defaults to `False`): Whether to use XPU if available. T)parentsexist_okz Configuration already exists at zZ, will not override. Run `accelerate config` manually or pass a different `save_location`.F)r fp16bf16fp8zL`mixed_precision` should be one of 'no', 'fp16', 'bf16', or 'fp8'. Received LOCAL_MACHINE)compute_environmentmixed_precision num_processesuse_cpur MULTI_GPUdistributed_typeNO MULTI_XPU MULTI_NPUrdebug)rparentmkdirexistsprintlower ValueErrortorchcuda is_available device_countrxpurnpur to_json_file)rr r pathconfignum_gpusnum_xpusnum_npuss l/var/www/html/backtest/airagagent/rag_env/lib/python3.12/site-packages/accelerate/commands/config/default.pywrite_basic_configr3s  DKKdT2 {{} .}o>X Y %++-O;;Z[jZk l   /*F zz ::**,"*!y a<)4F% &)-F% &  99))+"*!y a<)4F% &)-F% &  99))+"*!y a<)4F% &)-F% & y"#%)!"F7O  $V $F  Kc|jd|tt}|jdtdd|jdgdt d d |j t |S) Ndefault)rhelpformatter_classz --config_fileaTThe path to use to store the config file. Will default to a file named default_config.yaml in the cache location, which is the content of the environment `HF_HOME` suffixed with 'accelerate', or if you don't have such an environment variable, your cache directory ('~/.cache' or the content of `XDG_CACHE_HOME`) suffixed with 'huggingface'.r )r6r7destz--mixed_precision)r rrzWhether or not to use mixed precision training. Choose between FP16 and BF16 (bfloat16) training. BF16 training is only supported on Nvidia Ampere GPUs and PyTorch 1.10 or later.r )choicestyper7r6)func) add_parser descriptionr add_argumentr str set_defaultsdefault_config_command)parserrs r2default_command_parserrD_s   y' ]t  uF ( "   & [ 34 Mr4cft|j|j}|rtd|yy)Nz"accelerate configuration saved at )r3rr r#)args config_files r2rBrBzs2$T%9%94;M;MNK 2;-@Ar4)pathlibrr&utilsrr config_argsrr config_utilsr r>r@boolr3rDrBrr4r2rMsI" 7@1W (,BZlq?C?ei?D6Br4