Ë
    S^(h  ã                   óª   — d dl 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	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_safetensors_availableÚis_tf_availableÚis_torch_availableÚis_torch_hpu_availableÚis_torch_npu_availableé   )ÚBaseTransformersCLICommandc                 ó   — t        «       S ©N)ÚEnvironmentCommand)Ú_s    úW/var/www/skyplay_api_hub/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_subcommand0   sT   € à ×+Ñ+¨EÓ2ˆØ×$Ñ$Ô*>Ð$Ô?Ø×$Ñ$Ø&ØØdð 	%ô 	
ð
 	×$Ñ$Ô*BÐ$ÕCr   Nc                 ó   — || _         y r   )Ú_accelerate_config_file)Úselfr   r   s      r   Ú__init__zEnvironmentCommand.__init__;   s
   € Ø'=ˆÕ$r   c                 óÜ  — d}t        «       rdd l}|j                  }n2t        j                  j                  d«      �dd l}|j                  › d�}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}t)        «       rXdd l}|j                  }|j,                  j/                  «       }|j0                  j/                  «       }t3        «       }t5        «       }d}d}t7        «       r+dd l}|j                  }	 |j:                  j=                  «       }d}tG        «       r@tI        jJ                  tM        jN                  «       «      5  dd l(}d d d «       j                  }d}d}d}d}tS        «       r^dd l*}dd l+}dd l,}|j                  }|j                  }|j                  }|jZ                  j\                  j_                  «       j`                  }tb        ta        j`                  «       ta        jd                  «       tf        j                  |› |› |› |› |› d|› d�|› d|› d�|› d|› d�|› |› ddœ}t)        «       r¬|r#d|d<   j,                  ji                  «       |d<   n‡r#d|d<   j0                  ji                  «       |d<   nbr#d|d<   jj                  ji                  «       |d<   n=r;d|d<   jl                  ji                  «       |d<   |jb                  jn                  |d<   tq        d«       tq        | js                  |«      «       |S c c}
}	w # t>        $ r( tA        |jB                  jE                  d«      «      }Y �Œ.w xY w# 1 sw Y   �ŒüxY w)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	   r,   r   Ú	importlibÚutilÚ	find_specr   Ú
accelerateÚaccelerate.commands.configr-   r.   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_versionr,   Úaccelerate_versionÚaccelerate_configÚaccelerate_config_strr9   r-   r.   ÚpropÚvalÚ
pt_versionÚpt_cuda_availablerC   Ú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½  € Ø-ÐÜ#Ô%Ûà"-×"9Ñ"9ÑÜ�^‰^×%Ñ% mÓ4Ð@Ûà%0×%<Ñ%<Ð$=Ð=pÐ"qÐà,ÐØ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ÐØ$Ÿy™y×5Ñ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Ø)3¨°BÐ7HÐ6IÈÐ&KØ,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Ó$Ô%àˆùói \øô2 "ò Qä$(¨¯©×)HÑ)HÈÓ)OÓ$PÓ!ðQú÷!ñ !ús$   Ã1N'
ÆN- Ç"O!Î--OÏOÏ!O+c           
      ó„   — dj                  | j                  «       D ��cg c]  \  }}d|› d|› �‘Œ c}}«      dz   S c c}}w )Nr/   z- r0   )rA   rB   )Údrg   rh   s      r   rb   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*   rw   rb   © r   r   r   r   /   s@   „ ØðD Nò Dó ðDó>òkðZ ñQó ñQr   r   )rN   Úimportlib.utilr6   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…      sQ   ðó  Û Û 	Û 	Û Ý #ã å %Ý ;÷÷ ñ õ )ò ò;ô~QÐ3õ ~Qr   