
    <hp                         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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is_torch_xpu_available   )BaseTransformersCLICommandc                     [        5       $ N)EnvironmentCommand)_s    Q/var/www/html/shao/venv/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   0   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_subcommand1   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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      (       a9  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[1        5       n[3        5       n[5        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[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Rj                  Ri                  5       US'   OlW(       a#  SUS'   WRl                  Ri                  5       US'   OBW(       a;  SUS'   WRn                  Ri                  5       US'   URb                  Rp                  US'   [s        S5        [s        U Ru                  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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.
);r	   r1   r   	importlibutil	find_specr   
accelerateaccelerate.commands.configr2   r3   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-   safetensors_versionr1   accelerate_versionaccelerate_configaccelerate_config_strrB   r2   r3   propval
pt_versionpt_cuda_availablept_acceleratorrL   pt_xpu_availablept_npu_availablept_hpu_available
tf_versiontf_cuda_availabletfdeepspeed_versionrZ   flax_versionjax_versionjaxlib_versionjax_backendr[   r\   r]   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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$   P
P "Q.P?>P?
Q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 )Nr4   z- r5   )rJ   rK   )drp   rq   s      r   rk   EnvironmentCommand.format_dict   s=    yyaggiHiBtfBse,iHIDPPHs   =
r+   )returnN)__name__
__module____qualname____firstlineno__staticmethodr   r(   r.   r   rk   __static_attributes__ r   r   r   r   0   s@    DN D D>un Q Qr   r   )rV   importlib.utilr?   rX   rD   ra   argparser   rd    r   rb   integrations.deepspeedr   utilsr   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   <module>r      sQ       	 	  #  % ;	 	 	 ) ;HQ3 HQr   