U
    0-e                     @   s   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dZdd	 Zed
ddZedkree dS )    N)__version__)default_config_fileload_config_from_file   )is_npu_availableis_xpu_availablec                 C   sF   | d k	r|  d}n
td}|jdd dd | d k	rB|jtd |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)Z
subparsersparser r   X/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/accelerate/commands/env.pyenv_command_parser   s    
  r   c              	   C   s  t j}t j }t }t }d}| jd k	s6tj	t
rDt| j }tt t tj| d| dt|t|t jd ddd}|rt j |d< td	 td
dd | D  t| jd krdnd t|trd
dd | D nd| }t| ||d< |S )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

c                 S   s    g | ]\}}d | d| qS )z- : r   .0propvalr   r   r   
<listcomp>G   s     zenv_command.<locals>.<listcomp>z- `Accelerate` default config:z- `Accelerate` config passed:c                 S   s    g | ]\}}d | d| qS )z	- r   r   r   r   r   r   r   K   s     	z`Accelerate` configs)torchr   cudaZis_availabler   r   config_fileospathisfiler   r   to_dictversionplatformpython_versionnpstrpsutilZvirtual_memorytotalZget_device_nameprintjoinitems
isinstancedict)argsZ
pt_versionZpt_cuda_availableZpt_xpu_availableZpt_npu_availableZaccelerate_configinfoZaccelerate_config_strr   r   r   r   .   s8    

r   )returnc                  C   s   t  } |  }t| dS )Nr   )r   
parse_argsr   )r   r3   r   r   r   mainV   s    r7   __main__)N)r   r#   r(   numpyr*   r,   r    Z
accelerater   r'   Zaccelerate.commands.configr   r   utilsr   r   r   r   intr7   __name__
SystemExitr   r   r   r   <module>   s   
(