
    ;i                         S SK r S SKrS SKJr  SSKJr  SSKJrJrJ	r	J
r
  SSKJrJrJr  SSKJr  SrS	 rSS
 jrS rS r\S:X  a  \" 5         gg)    N)ComputeEnvironment   )get_cluster_input)	cache_dirdefault_config_filedefault_yaml_config_fileload_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 machinec                      [        SSS/[        5      n U [        R                  :X  a  [	        5       nU$ [        5       nU$ )Nz-In which compute environment are you running?zThis machinezAWS (Amazon SageMaker))r   r   r   AMAZON_SAGEMAKERr   r   )compute_environmentconfigs     q/home/dmtnaga/Documents/work/airagagent/rag_env/lib/python3.13/site-packages/accelerate/commands/config/config.pyget_user_inputr      sK    &7	12$
 0AAA$& M #$M    c                     U b  U R                  S[        S9nO[        R                  " S[        S9nUR	                  SS SS9  U b  UR                  [        S9  U$ )Nr   )descriptionzAccelerate config commandz--config_fileaT  The 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'.)defaulthelp)func)
add_parserr   argparseArgumentParseradd_argumentset_defaultsconfig_command)
subparsersparsers     r   config_command_parserr"   ,   sm    &&x[&I(()DR]^
"	  	 0Mr   c                 j   [        5       nU R                  b  U R                  nOH[        R                  R	                  [
        5      (       d  [        R                  " [
        5        [        nUR                  S5      (       a  UR                  U5        OUR                  U5        [        SU 35        g )Nz.jsonz"accelerate configuration saved at )r   config_fileospathisdirr   makedirsr   endswithto_json_fileto_yaml_fileprint)argsr   r$   s      r   r   r   B   s    F#&&ww}}Y''KK	".G$$K(K(	.{m
<=r   c                  N    [        5       n U R                  5       n[        U5        g N)r"   
parse_argsr   )r!   r-   s     r   mainr1   R   s     "$FD4r   __main__r/   )r   r%   accelerate.utilsr   clusterr   config_argsr   r   r   r	   config_utilsr
   r   r   	sagemakerr   r   r   r"   r   r1   __name__ r   r   <module>r:      sR   "  	 / & h h P P * p
,>  zF r   