
    #jy                         d 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ZddlZddl	m
Z
 ddlmZ ddlmZmZmZmZmZ 	 ddeedz   ej        d	
          f         ddfdZddZdedefdZdS )zContains commands to print information about the environment and version.

Usage:
    transformers env
    transformers version
    N)	Annotated   )__version__)is_deepspeed_available)is_accelerate_availableis_torch_availableis_torch_hpu_availableis_torch_npu_availableis_torch_xpu_availableaccelerate_config_filezQThe accelerate config file to use for the default values in the launching script.)helpreturnc                    ddl }t          |d          r|j        nd}d}dx}}t                      rddl}ddlm}m} |j        }| t          j	        
                    |          r ||                                           }t          |t                    r1d                    d	 |                                D                       nd
| }d}	d}
d}t!                      raddl}|j        }	|j                                        }
t)                      }t+                      }t-                      }|
rd}n|rd}n	|rd}n|rd}d}t/                      rHt1          j        t5          j                              5  ddl}ddd           n# 1 swxY w Y   |j        }t          t;          j                    t;          j                    t>          j        | | | | |	 d| ddd
}t!                      r|
r"d|d<   |j                                         |d<   n|r"d|d<   |j!                                         |d<   n|r2t          |d          r"d|d<   |j"                                         |d<   nW|rUt          |d          rEd|d<   |j#                                         |d<   t          |j$        d          r|j$        j%        |d<   tM          d            tM          tO          |                     |S )!z(Print information about the environment.r   Nr   unknownznot installedz	not found)default_config_fileload_config_from_file
c                 $    g | ]\  }}d | d| S )z	- :  .0propvals      a/lsinfo/ai/hellotax_ai/base_platform/venv/lib/python3.11/site-packages/transformers/cli/system.py
<listcomp>zenv.<locals>.<listcomp>B   s-    WWW	c+d++c++WWW    	NACUDAXPUNPUHPUz ()z	<fill in>)
z`transformers` versionPlatformzPython versionzHuggingface_hub versionzSafetensors versionzAccelerate versionzAccelerate configzDeepSpeed versionzPyTorch version (accelerator?)z/Using distributed or parallel set-up in script?zUsing GPU in script?zGPU typezUsing XPU in script?zXPU typehpuzUsing HPU in script?zHPU typenpuzUsing NPU in script?zNPU typecannzCANN versionzV
Copy-and-paste the text below in your GitHub issue and FILL OUT the two last points.
)(safetensorshasattrr   r   
accelerateaccelerate.commands.configr   r   ospathisfileto_dict
isinstancedictjoinitemsr   torchcudais_availabler   r
   r	   r   
contextlibredirect_stdoutioStringIO	deepspeedplatformpython_versionhuggingface_hubget_device_namexpur&   r'   versionr(   print_format_dict)r   r)   safetensors_versionaccelerate_versionaccelerate_configaccelerate_config_strr+   r   r   
pt_versionpt_cuda_availablept_acceleratorr5   pt_xpu_availablept_npu_availablept_hpu_availabledeepspeed_versionr<   infos                      r   envrQ   )   s     6=[-5X5Xg+11^g(0;;-   
YYYYYYYY'3!-@S1T1T- 5 56L M M U U W W +T22*DIIWW=N=T=T=V=VWWWXXX)')) 	 !JN #&
!J3355133133133 	##NN 	#"NN 	#"NN 	#"N' 2'66 	 		 	 	 	 	 	 	 	 	 	 	 	 	 	 	%1 #.%''"133#2#>"57!35 57 13-7*L*L>*L*L*L;F D  : 	:+6D'($z99;;D 
	:+6D'($y88::D 	:'%"7"7 	:+6D'($y88::D 	:'%"7"7 	:+6D'($y88::Du}f-- :',}'9^$	
deee	,t

Ks   8F		FFc                  .    t          t                     dS )zPrint CLI version.N)rC   r   r   r   r   rB   rB      s    	+r   dc                 j    d                     d |                                 D                       dz   S )Nr   c                 $    g | ]\  }}d | d| S )z- r   r   r   s      r   r   z _format_dict.<locals>.<listcomp>   s-    DDDYT3(4((3((DDDr   )r3   r4   )rS   s    r   rD   rD      s0    99DD!''))DDDEELLr   )N)r   N)__doc__r8   r:   r-   r=   typingr   r?   typer r   integrations.deepspeedr   utilsr   r   r	   r
   r   strArgumentrQ   rB   r2   rD   r   r   r   <module>r^      sn        				 				                   ; ; ; ; ; ;              		Y Y%d
oppp	rY
 
Y Y Y Yx   
MD MS M M M M M Mr   