
    ;i                         S SK r S SKrS SKrS SKrS SKrS SKrS SKrS SKJ	r
  S SKJrJr  SSKJrJrJrJrJr  SS jrS rS\4S	 jr\S
:X  a  \" \" 5       5      eg)    N)__version__)default_config_fileload_config_from_file   )is_mlu_availableis_musa_availableis_npu_availableis_sdaa_availableis_xpu_availablec                     U b  U R                  S5      nO[        R                  " S5      nUR                  SS SS9  U b  UR	                  [
        S9  U$ )NenvzAccelerate env commandz--config_filezFThe config file to use for the default values in the launching script.)defaulthelp)func)
add_parserargparseArgumentParseradd_argumentset_defaultsenv_command)
subparsersparsers     g/home/dmtnaga/Documents/work/airagagent/rag_env/lib/python3.13/site-packages/accelerate/commands/env.pyenv_command_parserr       sb    &&u-(()AB
,t   -M    c                 (   [         R                  n[         R                  R                  5       n[	        5       n[        5       n[        5       n[        5       n[        5       nSnU(       a  SnO1U(       a  SnO'U(       a  SnOU(       a  SnOU(       a  SnO	U(       a  SnSn	U R                  c(  [        R                  R                  [        5      (       a#  [        U R                  5      R                  5       n	S n
Sn[        R                   S	:X  a  S
S/n
O[        R                   S:X  a  SS/n
U
b2  ["        R$                  " U
S["        R&                  S9R)                  5       n[*        [,        R,                  " 5       U[,        R.                  " 5       [0        R                  U U[2        R4                  " 5       R6                  S-  S S3S.nU(       a"  [         R                  R9                  5       US'   OU(       a"  [         R:                  R9                  5       US'   OU(       a"  [         R<                  R9                  5       US'   OvU(       a"  [         R>                  R9                  5       US'   OMU(       a"  [         R@                  R9                  5       US'   O$U(       a  [         R*                  RB                  US'   [E        S5        [E        SRG                  URI                  5        VVs/ s H  u  pSU SU 3PM     snn5      5        [E        U R                  c  SOS5        [K        U	[L        5      (       a:  SRG                  U	RI                  5        VVs/ s H  u  pS U SU 3PM     snn5      OS!U	 3n[E        U5        XS"'   U$ s  snnf s  snnf )#NzN/ACUDAXPUMLUSDAAMUSANPUz	Not foundntwhere
accelerateposixwhichT)textstderri   @z.2fz GB)z`Accelerate` versionPlatformz`accelerate` bash locationzPython versionzNumpy versionzPyTorch versionzPyTorch acceleratorz
System RAMzGPU typezXPU typezMLU typez	SDAA typez	MUSA typezCANN versionz4
Copy-and-paste the text below in your GitHub issue

z- z: z- `Accelerate` default config:z- `Accelerate` config passed:z	- 	z`Accelerate` configs)'torchr   cudais_availabler   r   r
   r   r	   config_fileospathisfiler   r   to_dictname
subprocesscheck_outputSTDOUTstripversionplatformpython_versionnppsutilvirtual_memorytotalget_device_namexpumlusdaamusacannprintjoinitems
isinstancedict)args
pt_versionpt_cuda_availablept_xpu_availablept_mlu_availablept_sdaa_availablept_musa_availablept_npu_availableacceleratoraccelerate_configcommandbash_locationinfopropvalaccelerate_config_strs                   r   r   r   /   s   ""J

//1')'))+)+')K					##rww~~6I'J'J1$2B2BCKKM GM	ww$L)	G	L)"//d:K\K\]cce '%%'&3"113(\*..066@ESI	D  ::557Z	 99446Z	 99446Z	!JJ668[	!JJ668[	$}}11^	
BC	$))FIDr$r#'F
GH	d.>.>.F
*Lkl '.. 			9J9P9P9RS9RIDT$r#'9RST#$% 
 

 #4	 K G Ts   )NN
returnc                  N    [        5       n U R                  5       n[        U5        g)Nr   )r   
parse_argsr   )r   rL   s     r   mainr_   {   s#    !FDr   __main__)N)r   r1   r;   r6   numpyr=   r>   r-   r%   r   r:   accelerate.commands.configr   r   utilsr   r   r	   r
   r   r   r   intr_   __name__
SystemExit r   r   <module>rh      s\   "  	      - Q n nIXc  z
TV
 r   