
    eTh                         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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_safetensors_availableis_tf_availableis_torch_availableis_torch_hpu_availableis_torch_npu_available   )BaseTransformersCLICommandc                     [        5       $ N)EnvironmentCommand)_s    Q/var/www/auris/envauris/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_subcommand0   sT     ++E2$$*>$?$$&d 	% 	

 	$$*B$Cr   Nc                     Xl         g r   _accelerate_config_file)selfr   r   s      r   __init__EnvironmentCommand.__init__;   s    '=$r   c                 x   Sn[        5       (       a  SS KnUR                  nO3[        R                  R                  S5      b  SS KnUR                   S3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[)        5       (       aX  SS KnUR                  nUR,                  R/                  5       nUR0                  R/                  5       n[3        5       n[5        5       nSnSn[7        5       (       a+  SS KnUR                  n UR:                  R=                  5       nSn[G        5       (       aB  [H        RJ                  " [L        RN                  " 5       5         SS K(nS S S 5        WR                  nSnSnSnSn[S        5       (       a^  SS K*nSS K+nSS K,nUR                  nUR                  nUR                  nURZ                  R\                  R_                  5       R`                  n[b        [`        R`                  " 5       [`        Rd                  " 5       [f        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,                  Ri                  5       US'   OW(       a#  SUS'   WR0                  Ri                  5       US'   OlW(       a#  SUS'   WRj                  Ri                  5       US'   OBW(       a;  SUS'   WRl                  Ri                  5       US'   URb                  Rn                  US'   [q        S5        [q        U Rs                  U5      5        U$ s  sn
n	f ! [>         a(    [A        URB                  RE                  S5      5      n GNVf = f! , (       d  f       GN"= f)Nznot installedr   safetensorsz3 but is ignored because of PyTorch version too old.z	not found)default_config_fileload_config_from_file
z	- : 	NAGPUz ()z	<fill in>)z`transformers` versionPlatformzPython versionzHuggingface_hub versionzSafetensors versionzAccelerate versionzAccelerate configzDeepSpeed versionzPyTorch version (GPU?)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.
):r	   r0   r   	importlibutil	find_specr   
accelerateaccelerate.commands.configr1   r2   r+   ospathisfileto_dict
isinstancedictjoinitemsr   torchcudais_availablexpur   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hpunpucannprintformat_dict)r,   safetensors_versionr0   accelerate_versionaccelerate_configaccelerate_config_strr=   r1   r2   propval
pt_versionpt_cuda_availablerG   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   -#%%"-"9"9^^%%m4@%0%<%<$==p"q,4??"$$]!+!7!7++7277>>J];^;^$9$:V:V$W$_$_$a! /66 		ARAXAXAZ[AZIDT$r#/AZ[\+,- " %
 **J %

 7 7 9$yy5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#6)3B7H6I&K,6<r:K9LA)N0<~R}A-N)]!/ 0?J
   /:+,#(::#=#=#?Z !/:+,#(99#<#<#>Z !/:+,#(99#<#<#>Z !/:+,#(99#<#<#>Z ',}}'9'9^$hidt$%i \2 " Q$()H)H)O$P!Q ;:s$   O/
6O5 
P*5.P'&P'*
P9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 )Nr3   z- r4   )rE   rF   )drk   rl   s      r   rf   EnvironmentCommand.format_dict   s=    yyaggiHiBtfBse,iHIDPPHs   >
r*   )returnN)__name__
__module____qualname____firstlineno__staticmethodr   r'   r-   r{   rf   __static_attributes__ r   r   r   r   /   s@    DN D D>kZ Q Qr   r   )rR   importlib.utilr:   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      sQ       	 	  #  % ;   ) ;~Q3 ~Qr   