ó
    Eñ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_inputzŸLaunches 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     Ú^/home/mande/repo/quber/.venv/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‰ä×(Ò(Ð)DÔR]Ñ^ˆà
×ÑØØð"ð	 ñ 	ð ÑØ×Ñ¤ÐÑ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Ø×ÑÑ#Ø×&Ñ&‰ä�w‰w�}‰}œY×'Ñ'Ü�KŠKœ	Ô"Ü.ˆà×Ñ˜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   