
    
9j                         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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_neuron_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     Q/home/wildlama/miniconda3/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[        5       nSn	U(       a  Sn	O;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[.        R0                  " 5       [2        R                  U U	[4        R6                  " 5       R8                  S-  S S3S.nU(       a"  [         R                  R;                  5       US'   OU(       a"  [         R<                  R;                  5       US'   OU(       a"  [         R>                  R;                  5       US'   OU(       a"  [         R@                  R;                  5       US'   OvU(       a"  [         RB                  R;                  5       US'   OMU(       a"  [         RD                  R;                  5       US'   O$U(       a  [         R,                  RF                  US'   [I        S5        [I        SRK                  URM                  5        VVs/ s H  u  pSU SU 3PM     snn5      5        [I        U R                  c  S OS!5        [O        U
[P        5      (       a:  SRK                  U
RM                  5        VVs/ s H  u  pS"U SU 3PM     snn5      OS#U
 3n[I        U5        XS$'   U$ s  snnf s  snnf )%NzN/ACUDAXPUMLUSDAAMUSANPUNEURONz	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NEURON 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
   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neuron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pt_neuron_availableacceleratoraccelerate_configcommandbash_locationinfopropvalaccelerate_config_strs                    r   r   r   6   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[	#ll::<]	$}}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   &OO
returnc                  N    [        5       n U R                  5       n[        U5        g)Nr   )r   
parse_argsr   )r   rO   s     r   mainrc      s#    !FDr   __main__)N)r   r3   r=   r8   numpyr?   r@   r/   r'   r   r<   accelerate.commands.configr   r   utilsr   r   r	   r
   r   r   r   r   intrc   __name__
SystemExit r   r   <module>rl      s`   "  	      - Q Nbc  z
TV
 r   