U 0-e @sddlZddlZddlmZddlmZddlmZmZm Z m Z ddl m Z m Z mZddlmZdZd d Zdd d Zd dZddZedkredS)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 machinecCs,tdddgt}|tjkr"t}nt}|S)Nz-In which compute environment are you running?z This machinezAWS (Amazon SageMaker))r r rZAMAZON_SAGEMAKERr r)Zcompute_environmentconfigrb/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/accelerate/commands/config/config.pyget_user_inputs rcCsN|dk r|jdtd}ntjdtd}|jdddd|dk rJ|jtd|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)Z subparsersparserrrrconfig_command_parser,s  rcCsdt}|jdk r|j}ntjts.ttt}|drH| |n | |t d|dS)Nz.jsonz"accelerate configuration saved at ) r config_fileospathisdirrmakedirsrendswithZ to_json_fileZ to_yaml_fileprint)argsr rrrrrBs      rcCst}|}t|dS)N)r parse_argsr)rr$rrrmainRsr&__main__)N)rrZaccelerate.utilsrZclusterrZ config_argsrrrrZ config_utilsr r r Z sagemakerr rrrrr&__name__rrrrs