Ë
    îÍ:j  ã                   ó¢   — 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mZ	 ddl
mZ ddlmZmZmZmZmZmZmZ ddlmZ d	„ Zd
„ Z G d„ de«      Zy)é    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                 ó   — t        «       S ©N)ÚEnvironmentCommand)Ú_s    ún/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/transformers/commands/env.pyÚinfo_command_factoryr   &   s   € ÜÓÐó    c                 ó,   — t        | j                  «      S r   )r   Úaccelerate_config_file)Úargss    r   Údownload_command_factoryr   *   s   € Ü˜d×9Ñ9Ó:Ð:r   c                   ó@   — e Zd Zedefd„«       Zdd„Zd„ Zed„ «       Zy)r   Úparserc                 ó¦   — | j                  d«      }|j                  t        ¬«       |j                  dd d¬«       |j                  t        ¬«       y )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_subcommandz&EnvironmentCommand.register_subcommand/   sT   € à ×+Ñ+¨EÓ2ˆØ×$Ñ$Ô*>Ð$Ô?Ø×$Ñ$Ø&ØØdð 	%ô 	
ð
 	×$Ñ$Ô*BÐ$ÕCr   Nc                 ó   — || _         y r   )Ú_accelerate_config_file)Úselfr   r   s      r   Ú__init__zEnvironmentCommand.__init__:   s
   € Ø'=ˆÕ$r   c                 óh  — dd l }|j                  }d}dx}}t        «       r°dd l}ddlm}m} |j                  }| j                  €t        j                  j                  |«      r  || j                  «      j                  «       }t        |t        «      r8dj                  |j                  «       D �	�
cg c]  \  }	}
d|	› d|
› �‘Œ c}
}	«      nd|› �}d}d	}d	}t!        «       r[dd l}|j                  }|j$                  j'                  «       }t)        «       }t+        «       }t-        «       }|rd
}n|rd}n	|rd}n|rd}d}d	}t/        «       r+dd l}|j                  }	 |j2                  j5                  «       }d}t?        «       r@tA        jB                  tE        jF                  «       «      5  dd l$}d d d «       j                  }d}d}d}d	}tK        «       r^dd l&}dd l'}dd l(}|j                  }|j                  }|j                  }|jR                  jT                  jW                  «       jX                  }tZ        tY        jX                  «       tY        j\                  «       t^        j                  |› |› |› |› |› d|› d�|› d|› d�|› d|› d�|› |› ddœ}t!        «       r¬|r#d|d<   j$                  ja                  «       |d<   n‡r#d|d<   jb                  ja                  «       |d<   nbr#d|d<   jd                  ja                  «       |d<   n=r;d|d<   jf                  ja                  «       |d<   |jZ                  jh                  |d<   tk        d«       tk        | jm                  |«      «       |S c c}
}	w # t6        $ r( t9        |j:                  j=                  d«      «      }Y �Œ.w xY w# 1 sw Y   �ŒüxY w)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.configr,   r-   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)   r9   Úsafetensors_versionÚaccelerate_versionÚaccelerate_configÚaccelerate_config_strr:   r,   r-   ÚpropÚvalÚ
pt_versionÚpt_cuda_availableÚpt_acceleratorrD   Úpt_xpu_availableÚpt_npu_availableÚpt_hpu_availableÚ
tf_versionÚtf_cuda_availableÚtfÚdeepspeed_versionrR   Úflax_versionÚjax_versionÚjaxlib_versionÚjax_backendrS   rT   rU   Úinfos                                  r   ÚrunzEnvironmentCommand.run=   sŸ  € Ûà)×5Ñ5Ðà,ÐØ4?Ð?ÐÐ1ä"Ô$Ûß]à!+×!7Ñ!7Ðà×+Ñ+Ð7¼2¿7¹7¿>¹>ÐJ]Ô;^Ù$9¸$×:VÑ:VÓ$W×$_Ñ$_Ó$aÐ!ô Ð/´Ô6ð —	‘	ÐAR×AXÑAXÓAZ×[±I°D¸#˜T $  r¨#¨Ò/Ó[Ô\àÐ+Ð,Ð-ð "ð %ˆ
Ø ÐØˆÜÔÛà×*Ñ*ˆJØ %§
¡
× 7Ñ 7Ó 9ÐÜ5Ó7ÐÜ5Ó7ÐÜ5Ó7Ðá Ø!'‘Ù!Ø!&‘Ù!Ø!&‘Ù!Ø!&�à$ˆ
Ø ÐÜÔÛ#àŸ™ˆJðQà$&§G¡G×$<Ñ$<Ó$>Ð!ð
 ,ÐÜ!Ô#ä×+Ñ+¬B¯K©K«MÓ:ñ !Û ÷!à )× 5Ñ 5Ðà&ˆØ%ˆØ(ˆØˆÜÔÛÛÛà×+Ñ+ˆLØŸ/™/ˆKØ#×/Ñ/ˆNØŸ'™'×,Ñ,×8Ñ8Ó:×CÑCˆKô '.Ü ×)Ñ)Ó+Ü&×5Ñ5Ó7Ü'6×'BÑ'BØ&9Ð%:Ø%7Ð$8Ø$9Ð#:Ø$5Ð#6Ø1;°¸B¸~Ð>NÈaÐ.PØ,6¨<°rÐ:KÐ9LÈAÐ)NØ0<¨~¸RÀ¸}ÈAÐ-NØ)˜]Ø!/Ð 0Ø?Jñ
ˆô  ÔÙ Ø/:�Ð+Ñ,Ø#(§:¡:×#=Ñ#=Ó#?��ZÒ Ù!Ø/:�Ð+Ñ,Ø#(§9¡9×#<Ñ#<Ó#>��ZÒ Ù!Ø/:�Ð+Ñ,Ø#(§9¡9×#<Ñ#<Ó#>��ZÒ Ù!Ø/:�Ð+Ñ,Ø#(§9¡9×#<Ñ#<Ó#>��ZÑ Ø',§}¡}×'9Ñ'9��^Ñ$äÐhÔiÜˆd×Ñ˜tÓ$Ô%àˆùó} \øôF "ò Qä$(¨¯©×)HÑ)HÈÓ)OÓ$PÓ!ðQú÷!ñ !ús$   Â2M-
ÅM3 Æ(N'Í3-N$Î#N$Î'N1c           
      ó„   — dj                  | j                  «       D ��cg c]  \  }}d|› d|› �‘Œ c}}«      dz   S c c}}w )Nr.   z- r/   )rB   rC   )Údrh   ri   s      r   rc   zEnvironmentCommand.format_dict¯   s;   € à�y‰y¸a¿g¹g»i×H±°°s˜B˜t˜f B s eÒ,ÓHÓIÈDÑPÐPùÓHs   Ÿ<
)ÚreturnN)	Ú__name__Ú
__module__Ú__qualname__Ústaticmethodr   r&   r*   ry   rc   © r   r   r   r   .   s@   „ ØðD Nò Dó ðDó>òpðd ñQó ñQr   r   )rN   rP   r<   rY   Úargparser   r\   Ú r   rZ   Ú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   