g ddlZddlZddlmZddlmZddlmZmZm Z m Z ddl m Z m Z mZddlmZdZd Zdd Zd Zd Zed kr edSdS)N)ComputeEnvironment)get_cluster_input) cache_dirdefault_config_filedefault_yaml_config_fileload_config_from_file) _ask_field _ask_options_convert_compute_environment)get_sagemaker_inputzLaunches a series of prompts to create and save a `default_config.yaml` configuration file for your training system. Should always be ran first on your machinectdddgt}|tjkrt }nt }|S)Nz-In which compute environment are you running?z This machinezAWS (Amazon SageMaker))r r rAMAZON_SAGEMAKERr r)compute_environmentconfigs ]/var/www/html/ai-engine/env/lib/python3.11/site-packages/accelerate/commands/config/config.pyget_user_inputrsO&7 12$ 0AAA$&&"$$ Mc||dt}ntjdt}|ddd||t |S)Nr) descriptionzAccelerate config commandz --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'.)defaulthelp)func) add_parserrargparseArgumentParser add_argument set_defaultsconfig_command) subparsersparsers rconfig_command_parserr",s&&x[&II()DR]^^^  "    000 Mrclt}|j|j}nDtjt stjt t}|dr| |n| |td|dS)Nz.jsonz"accelerate configuration saved at ) r config_fileospathisdirrmakedirsrendswith to_json_file to_yaml_fileprint)argsrr$s rrrBs   F #& w}}Y'' # K " " ". G$$)K((((K((( <{ < <=====rcht}|}t|dSN)r" parse_argsr)r!r-s rmainr1Rs2 " $ $F     D4r__main__r/)rr%accelerate.utilsrclusterr config_argsrrrr config_utilsr r r sagemakerr rrr"rr1__name__rrr:s" //////&&&&&&hhhhhhhhhhhhPPPPPPPPPP******p    , > > >   zDFFFFFr