§
    Ý˜ºhp  ã                   ó¶   — 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mZ ddl	mZ d	„ Zd
„ Z G d„ de¦  «        ZdS )é    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                 ó   — t          ¦   «         S ©N)ÚEnvironmentCommand)Ú_s    úa/var/www/html/services/rag-service/venv/lib/python3.11/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Ñ:Ô:Ð:r   c                   óN   — e Zd Zedefd„¦   «         Zdd„Zd„ Zed„ ¦   «         ZdS )	r   Úparserc                 óÌ   — |                       d¦  «        }|                     t          ¬¦  «         |                     dd d¬¦  «         |                     t          ¬¦  «         d S )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_subcommand1   ss   € à ×+Ò+¨EÑ2Ô2ˆØ×$Ò$Õ*>Ð$Ñ?Ô?Ð?Ø×$Ò$Ø&ØØdð 	%ñ 	
ô 	
ð 	
ð
 	×$Ò$Õ*BÐ$ÑCÔCÐCÐCÐCr   ÚreturnNc                 ó   — || _         d S r   )Ú_accelerate_config_file)Úselfr   r   s      r   Ú__init__zEnvironmentCommand.__init__<   s   € Ø'=ˆÔ$Ð$Ð$r   c                 ó¦  — d}t          ¦   «         rdd l}|j        }n-t          j                             d¦  «        �dd l}|j        › d�}d}dx}}t          ¦   «         r¦dd l}ddlm	}m
} |j        }| j        €t          j                             |¦  «        r" || j        ¦  «                             ¦   «         }t!          |t"          ¦  «        r1d                     d„ |                     ¦   «         D ¦   «         ¦  «        nd	|› �}d}	d
}
d
}t)          ¦   «         radd l}|j        }	|j                             ¦   «         }
t1          ¦   «         }t3          ¦   «         }t5          ¦   «         }|
rd}n|rd}n	|rd}n|rd}d}d
}t7          ¦   «         r]dd l}|j        }	 |j                             ¦   «         }n7# t>          $ r* tA          |j!         "                    d¦  «        ¦  «        }Y nw xY wd}tG          ¦   «         rHtI          j%        tM          j'        ¦   «         ¦  «        5  dd l(}d d d ¦  «         n# 1 swxY w Y   |j        }d}d}d}d
}tS          ¦   «         rDdd l*}dd l+}dd l,}|j        }|j        }|j        }|j-        j.         /                    ¦   «         j0        }tb          ta          j0        ¦   «         ta          j2        ¦   «         tf          j        |› |› |› |› |	› d|› d�|› d|› d�|› d|› d�|› |› ddœ}t)          ¦   «         rž|
r"d|d<   |j         4                    ¦   «         |d<   nz|r"d|d<   |j5         4                    ¦   «         |d<   nV|r"d|d<   |j6         4                    ¦   «         |d<   n2|r0d|d<   |j7         4                    ¦   «         |d<   |j1        j8        |d<   ts          d¦  «         ts          |  :                    |¦  «        ¦  «         |S )Nznot installedr   Úsafetensorsz3 but is ignored because of PyTorch version too old.z	not found)Údefault_config_fileÚload_config_from_fileú
c                 ó$   — g | ]\  }}d |› d|› �‘ŒS )z	- ú: © ©Ú.0ÚpropÚvals      r   ú
<listcomp>z*EnvironmentCommand.run.<locals>.<listcomp>V   s-   € Ð[Ð[Ð[±I°D¸#Ð/ $Ð/Ð/¨#Ð/Ð/Ð[Ð[Ð[r   ú	Ú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	   r.   r   Ú	importlibÚutilÚ	find_specr   Ú
accelerateÚaccelerate.commands.configr/   r0   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_versionr.   Úaccelerate_versionÚaccelerate_configÚaccelerate_config_strrF   r/   r0   Ú
pt_versionÚpt_cuda_availableÚpt_acceleratorrP   Úpt_xpu_availableÚpt_npu_availableÚpt_hpu_availableÚ
tf_versionÚtf_cuda_availableÚtfÚdeepspeed_versionr^   Úflax_versionÚjax_versionÚjaxlib_versionÚjax_backendr_   r`   ra   Úinfos                                r   ÚrunzEnvironmentCommand.run?   s  € Ø-ÐÝ#Ñ%Ô%ð 	rØÐÐÐà"-Ô"9ÐÐÝŒ^×%Ò% mÑ4Ô4Ð@ØÐÐÐà%0Ô%<Ð"qÐ"qÐ"qÐà,ÐØ4?Ð?ÐÐ1Ý"Ñ$Ô$ð 	ØÐÐÐØ]Ð]Ð]Ð]Ð]Ð]Ð]Ð]à!+Ô!7ÐàÔ+Ð7½2¼7¿>º>ÐJ]Ñ;^Ô;^Ð7Ø$9Ð$9¸$Ô:VÑ$WÔ$W×$_Ò$_Ñ$aÔ$aÐ!õ Ð/µÑ6Ô6ð.�—	’	Ð[Ð[ÐAR×AXÒAXÑAZÔAZÐ[Ñ[Ô[Ñ\Ô\Ð\à-Ð+Ð-Ð-ð "ð %ˆ
Ø ÐØˆÝÑÔð 	'ØˆLˆLˆLàÔ*ˆJØ %¤
× 7Ò 7Ñ 9Ô 9ÐÝ5Ñ7Ô7ÐÝ5Ñ7Ô7ÐÝ5Ñ7Ô7Ðà ð 'Ø!'��Ø!ð 'Ø!&��Ø!ð 'Ø!&��Ø!ð 'Ø!&�à$ˆ
Ø ÐÝÑÔð 		QØ#Ð#Ð#Ð#àœˆJðQà$&¤G×$<Ò$<Ñ$>Ô$>Ð!Ð!øÝ!ð Qð Qð Qå$(¨¬×)HÒ)HÈÑ)OÔ)OÑ$PÔ$PÐ!Ð!Ð!ðQøøøð ,ÐÝ!Ñ#Ô#ð 	6åÔ+­B¬K©M¬MÑ:Ô:ð !ð !Ø Ð Ð Ð ð!ð !ð !ñ !ô !ð !ð !ð !ð !ð !ð !øøøð !ð !ð !ð !à )Ô 5Ðà&ˆØ%ˆØ(ˆØˆÝÑÔð 	DØˆKˆKˆKØˆJˆJˆJØˆMˆMˆMàÔ+ˆLØœ/ˆKØ#Ô/ˆNØœ'Ô,×8Ò8Ñ:Ô:ÔCˆKõ '.Ý Ô)Ñ+Ô+Ý&Ô5Ñ7Ô7Ý'6Ô'BØ&9Ð#;Ø%7Ð"9Ø$9Ð!;Ø$5Ð!7Ø1;Ð.PÐ.P¸~Ð.PÐ.PÐ.PØ,6Ð)NÐ)NÐ:KÐ)NÐ)NÐ)NØ0<Ð-NÐ-NÀÐ-NÐ-NÐ-NØ)Ð+Ø!/Ð1Ø?Jð
ð 
ˆõ  ÑÔð 	:Ø ð :Ø/:�Ð+Ñ,Ø#(¤:×#=Ò#=Ñ#?Ô#?��ZÑ Ð Ø!ð 	:Ø/:�Ð+Ñ,Ø#(¤9×#<Ò#<Ñ#>Ô#>��ZÑ Ð Ø!ð :Ø/:�Ð+Ñ,Ø#(¤9×#<Ò#<Ñ#>Ô#>��ZÑ Ð Ø!ð :Ø/:�Ð+Ñ,Ø#(¤9×#<Ò#<Ñ#>Ô#>��ZÑ Ø',¤}Ô'9��^Ñ$åÐhÑiÔiÐiÝˆd×Ò˜tÑ$Ô$Ñ%Ô%Ð%àˆs$   ÆF1 Æ11G%Ç$G%ÈH.È.H2È5H2c                 ój   — d                      d„ |                      ¦   «         D ¦   «         ¦  «        dz   S )Nr1   c                 ó$   — g | ]\  }}d |› d|› �‘ŒS )z- r3   r4   r5   s      r   r9   z2EnvironmentCommand.format_dict.<locals>.<listcomp>¸   s-   € ÐHÐHÐH±°°sÐ,˜tÐ,Ð, sÐ,Ð,ÐHÐHÐHr   )rN   rO   )Úds    r   ro   zEnvironmentCommand.format_dict¶   s0   € à�yŠyÐHÐH¸a¿gºg¹i¼iÐHÑHÔHÑIÔIÈDÑPÐPr   )r(   N)	Ú__name__Ú
__module__Ú__qualname__Ústaticmethodr   r'   r,   rƒ   ro   r4   r   r   r   r   0   s…   € € € € € ØðD Nð Dð Dð Dñ „\ðDð>ð >ð >ð >ðuð uð uðn ðQð Qñ „\ðQð Qð Qr   r   )rZ   Úimportlib.utilrC   r\   rH   re   Úargparser   rh   Ú r   rf   Úintegrations.deepspeedr   Úutilsr   r   r	   r
   r   r   r   r   r   r   r   r   r4   r   r   ú<module>r�      s[  ðð  Ð Ð Ð Ø Ð Ð Ð Ø 	€	€	€	Ø 	€	€	€	Ø €€€Ø #Ð #Ð #Ð #Ð #Ð #à Ð Ð Ð à %Ð %Ð %Ð %Ð %Ð %Ø ;Ð ;Ð ;Ð ;Ð ;Ð ;ð	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð )Ð (Ð (Ð (Ð (Ð (ð ð  ð  ð;ð ;ð ;ðHQð HQð HQð HQð HQÐ3ñ HQô HQð HQð HQð HQr   