
    i                         S SK r S SKrS SKrS SKrS SKJr  S SKrSSKJr	  SSK
Jr  SSKJrJrJrJrJrJrJr  SSKJr  S	 rS
 r " S S\5      rg)    N)ArgumentParser   )__version__)is_deepspeed_available)is_accelerate_availableis_flax_availableis_tf_availableis_torch_availableis_torch_hpu_availableis_torch_npu_availableis_torch_xpu_available   )BaseTransformersCLICommandc                     [        5       $ N)EnvironmentCommand)_s    i/home/dmtnaga/Documents/work/airagagent/rag_env/lib/python3.13/site-packages/transformers/commands/env.pyinfo_command_factoryr   &   s        c                 ,    [        U R                  5      $ r   )r   accelerate_config_file)argss    r   download_command_factoryr   *   s    d99::r   c                   L    \ rS rSr\S\4S j5       rS	S jrS r\S 5       r	Sr
g)
r   .   parserc                     U R                  S5      nUR                  [        S9  UR                  SS SS9  UR                  [        S9  g )Nenv)funcz--accelerate-config_filezQThe accelerate config file to use for the default values in the launching script.)defaulthelp)
add_parserset_defaultsr   add_argumentr   )r   download_parsers     r   register_subcommand&EnvironmentCommand.register_subcommand/   sT     ++E2$$*>$?$$&d 	% 	

 	$$*B$Cr   Nc                     Xl         g r   _accelerate_config_file)selfr   r   s      r   __init__EnvironmentCommand.__init__:   s    '=$r   c                     SS K nUR                  nSnS=pE[        5       (       a  SS KnSSKJnJn  UR                  nU R                  c$  [        R                  R                  U5      (       a   U" U R                  5      R                  5       n[        U[        5      (       a:  SR                  UR                  5        V	V
s/ s H  u  pSU	 SU
 3PM     sn
n	5      OSU 3nSnS	nS	n[!        5       (       ao  SS KnUR                  nUR$                  R'                  5       n[)        5       n[+        5       n[-        5       nU(       a  S
nOU(       a  SnOU(       a  SnO	U(       a  SnSnS	n[/        5       (       a+  SS KnUR                  n UR2                  R5                  5       nSn[?        5       (       aB  [@        RB                  " [D        RF                  " 5       5         SS K$nS S S 5        WR                  nSnSnSnS	n[K        5       (       a^  SS K&nSS K'nSS K(nUR                  nUR                  nUR                  nURR                  RT                  RW                  5       RX                  n[Z        [X        RX                  " 5       [X        R\                  " 5       [^        R                  U U U U U SU S3U SU S3U SU S3U U SS.n[!        5       (       a  U(       a#  SUS'   WR$                  Ra                  5       US'   OW(       a#  SUS'   WRb                  Ra                  5       US'   OlW(       a#  SUS'   WRd                  Ra                  5       US'   OBW(       a;  SUS'   WRf                  Ra                  5       US'   URZ                  Rh                  US'   [k        S5        [k        U Rm                  U5      5        U$ s  sn
n	f ! [6         a(    [9        UR:                  R=                  S5      5      n GNVf = f! , (       d  f       GN"= f)Nr   znot installedz	not found)default_config_fileload_config_from_file
z	- : 	NACUDAXPUNPUHPUGPUz ()z	<fill in>)z`transformers` versionPlatformzPython versionzHuggingface_hub versionzSafetensors versionzAccelerate versionzAccelerate configzDeepSpeed versionzPyTorch version (accelerator?)zTensorflow version (GPU?)zFlax version (CPU?/GPU?/TPU?)zJax versionzJaxLib versionz/Using distributed or parallel set-up in script?zUsing GPU in script?zGPU typezUsing XPU in script?zXPU typezUsing HPU in script?zHPU typezUsing NPU in script?zNPU typezCANN versionzV
Copy-and-paste the text below in your GitHub issue and FILL OUT the two last points.
)7safetensorsr   r   
accelerateaccelerate.commands.configr0   r1   r+   ospathisfileto_dict
isinstancedictjoinitemsr
   torchcudais_availabler   r   r   r	   
tensorflowtestis_gpu_availableAttributeErrorboolconfiglist_physical_devicesr   
contextlibredirect_stdoutioStringIO	deepspeedr   flaxjaxjaxliblib
xla_bridgeget_backendplatformversionpython_versionhuggingface_hubget_device_namexpuhpunpucannprintformat_dict)r,   r=   safetensors_versionaccelerate_versionaccelerate_configaccelerate_config_strr>   r0   r1   propval
pt_versionpt_cuda_availablept_acceleratorrH   pt_xpu_availablept_npu_availablept_hpu_available
tf_versiontf_cuda_availabletfdeepspeed_versionrV   flax_versionjax_versionjaxlib_versionjax_backendrW   rX   rY   infos                                  r   runEnvironmentCommand.run=   s   )55,4??"$$]!+!7!7++7277>>J];^;^$9$:V:V$W$_$_$a! /66 		ARAXAXAZ[AZIDT$r#/AZ[\+,- " %
 **J %

 7 7 9575757 !'!!&!!&!!&$
 #JQ$&GG$<$<$>!
 ,!##++BKKM:  ; ) 5 5&%(++L//K#//N'',,88:CCK '. ))+&557'6'B'B&9%:%7$8$9#:$5#61;B~>Na.P,6<r:K9LA)N0<~R}A-N)]!/ 0?J
   /:+,#(::#=#=#?Z !/:+,#(99#<#<#>Z !/:+,#(99#<#<#>Z !/:+,#(99#<#<#>Z ',}}'9'9^$hidt$%} \F " Q$()H)H)O$P!Q ;:s$   O

O	 O>	.O;:O;>
Pc           
          SR                  U R                  5        VVs/ s H  u  pSU SU 3PM     snn5      S-   $ s  snnf )Nr2   z- r3   )rF   rG   )drl   rm   s      r   rg   EnvironmentCommand.format_dict   s=    yyaggiHiBtfBse,iHIDPPHs   >
r*   )returnN)__name__
__module____qualname____firstlineno__staticmethodr   r'   r-   r}   rg   __static_attributes__ r   r   r   r   .   s@    DN D D>pd Q Qr   r   )rR   rT   r@   r]   argparser   r`    r   r^   integrations.deepspeedr   utilsr   r   r	   r
   r   r   r   r   r   r   r   r   r   r   <module>r      sN      	 	  #  % ;   ) ;CQ3 CQr   