
    9i                         d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ	 d dl
mZmZ ddlmZmZ ddZd Zdefd	Zed
k(  r e e             y)    N)__version__)default_config_fileload_config_from_file   )is_npu_availableis_xpu_availablec                     | | j                  d      }nt        j                  d      }|j                  dd d       | |j	                  t
               |S )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     a/var/www/html/backtest/airagagent/rag_env/lib/python3.12/site-packages/accelerate/commands/env.pyenv_command_parserr      sb    &&u-(()AB
,t   -M    c                    t         j                  }t         j                  j                         }t	               }t               }d}| j                  #t        j                  j                  t              r#t        | j                        j                         }t        t        j                         t        j                         t         j                  | d| dt#        |      t#        |      t%        j&                         j(                  dz  ddd}|r!t         j                  j+                         |d<   t-        d	       t-        d
j/                  |j1                         D cg c]  \  }}d| d|  c}}             t-        | j                  dnd       t3        |t4              r8d
j/                  |j1                         D cg c]  \  }}d| d|  c}}      nd| }	t-        |	       ||d<   |S c c}}w c c}}w )Nz	Not foundz ()i   @z.2fz GB)z`Accelerate` versionPlatformzPython versionzNumpy versionzPyTorch version (GPU?)zPyTorch XPU availablezPyTorch NPU availablez
System RAMzGPU typez4
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   config_fileospathisfiler   r   to_dictversionplatformpython_versionnpstrpsutilvirtual_memorytotalget_device_nameprintjoinitems
isinstancedict)
args
pt_versionpt_cuda_availablept_xpu_availablept_npu_availableaccelerate_configinfopropvalaccelerate_config_strs
             r   r   r   .   s   ""J

//1')')##rww~~6I'J1$2B2BCKKM !(%%'"113%/L3D2EQ"G!$%5!6!$%5!6..066B3GsK	D  ::557Z	
BC	$))FID#r$r#'F
GH	d.>.>.F
*Lkl '. 			9J9P9P9RSID#T$r#'ST#$% 
 

 #4D	 K G Ts   $G:	H 
returnc                  N    t               } | j                         }t        |       y)Nr   )r   
parse_argsr   )r   r4   s     r   mainrA   V   s#    !FDr   __main__)N)r   r"   r'   numpyr)   r+   r   
accelerater   r&   accelerate.commands.configr   r   utilsr   r   r   r   intrA   __name__
SystemExit r   r   <module>rK      sU   "  	     - Q 6%Pc  z
TV
 r   