Ë
    )täiê  ã                   ó¶   — d dl Z d dlZ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mZmZ ddlmZmZmZmZmZmZ dd„Zd„ Zdefd	„Zed
k(  r e e«       «      ‚y)é    N)Ú__version__)Údefault_config_fileÚload_config_from_fileé   )Úis_mlu_availableÚis_musa_availableÚis_neuron_availableÚis_npu_availableÚis_sdaa_availableÚis_xpu_availablec                 ó°   — | �| j                  d«      }nt        j                  d«      }|j                  dd d¬«       | �|j	                  t
        ¬«       |S )NÚenvzAccelerate env commandz--config_filezFThe config file to use for the default values in the launching script.)ÚdefaultÚhelp)Úfunc)Ú
add_parserÚargparseÚArgumentParserÚadd_argumentÚset_defaultsÚenv_command)Ú
subparsersÚparsers     úf/Volumes/fast/ai/experiments/MLX_z-image/.venv/lib/python3.12/site-packages/accelerate/commands/env.pyÚenv_command_parserr   '   sb   € ØÐØ×&Ñ& uÓ-‰ä×(Ñ(Ð)AÓBˆà
×ÑØ Ð,tð ô ð ÐØ×Ñ¤ÐÔ-Ø€Mó    c                 óð  — t         j                  }t         j                  j                  «       }t	        «       }t        «       }t        «       }t        «       }t        «       }t        «       }d}	|rd}	n|rd}	n|rd}	n|rd}	n|rd}	n	|rd}	n|rd}	d	}
| j                  €#t        j                  j                  t        «      r#t        | j                  «      j!                  «       }
d }d	}t        j"                  d
k(  rddg}nt        j"                  dk(  rddg}|�4t%        j&                  |dt$        j(                  ¬«      j+                  «       }t,        t/        j.                  «       |t/        j0                  «       t2        j                  |› |	t5        j6                  «       j8                  dz  d›d�dœ}|r"t         j                  j;                  «       |d<   nÓ|r"t         j<                  j;                  «       |d<   n¯|r"t         j>                  j;                  «       |d<   n‹|r"t         j@                  j;                  «       |d<   ng|r"t         jB                  j;                  «       |d<   nC|r"t         jD                  j;                  «       |d<   n|rt         j,                  jF                  |d<   tI        d«       tI        djK                  |jM                  «       D ��cg c]  \  }}d|› d|› �‘Œ c}}«      «       tI        | j                  €d nd!«       tO        |
tP        «      r8djK                  |
jM                  «       D ��cg c]  \  }}d"|› d|› �‘Œ c}}«      nd#|
› �}tI        |«       |
|d$<   |S c c}}w c c}}w )%NzN/AÚCUDAÚXPUÚMLUÚSDAAÚMUSAÚNPUÚNEURONz	Not foundÚntÚwhereÚ
accelerateÚposixÚwhichT)ÚtextÚstderri   @z.2fz GB)z`Accelerate` versionÚPlatformz`accelerate` bash locationzPython versionzNumpy versionzPyTorch versionzPyTorch acceleratorz
System RAMzGPU typezXPU typezMLU typez	SDAA typez	MUSA typezNEURON typezCANN versionz4
Copy-and-paste the text below in your GitHub issue
Ú
z- z: z- `Accelerate` default config:z- `Accelerate` config passed:z	- Ú	z`Accelerate` configs))Útorchr   ÚcudaÚis_availabler   r   r   r   r
   r	   Úconfig_fileÚosÚpathÚisfiler   r   Úto_dictÚnameÚ
subprocessÚcheck_outputÚSTDOUTÚstripÚversionÚplatformÚpython_versionÚnpÚpsutilÚvirtual_memoryÚtotalÚget_device_nameÚxpuÚmluÚsdaaÚmusaÚneuronÚcannÚprintÚjoinÚitemsÚ
isinstanceÚdict)ÚargsÚ
pt_versionÚpt_cuda_availableÚpt_xpu_availableÚpt_mlu_availableÚpt_sdaa_availableÚpt_musa_availableÚpt_npu_availableÚpt_neuron_availableÚacceleratorÚaccelerate_configÚcommandÚbash_locationÚinfoÚpropÚvalÚaccelerate_config_strs                    r   r   r   6   s  € Ü×"Ñ"€JÜŸ
™
×/Ñ/Ó1ÐÜ'Ó)ÐÜ'Ó)ÐÜ)Ó+ÐÜ)Ó+ÐÜ'Ó)ÐÜ-Ó/Ðà€KÙØ‰Ù	Ø‰Ù	Ø‰Ù	Ø‰Ù	Ø‰Ù	Ø‰Ù	Øˆà#Ðà×ÑÐ#¤r§w¡w§~¡~Ô6IÔ'JÜ1°$×2BÑ2BÓC×KÑKÓMÐð €GØ€MÜ	‡w�w�$‚Ø˜LÐ)‰Ü	�‰�GÒ	Ø˜LÐ)ˆØÐÜ"×/Ñ/°¸dÌ:×K\ÑK\Ô]×cÑcÓeˆä 'Ü×%Ñ%Ó'Ø&3Ü"×1Ñ1Ó3ÜŸ™Ø(˜\Ø*Ü×.Ñ.Ó0×6Ñ6¸Ñ@ÀÐEÀSÐIñ	€Dñ Ü Ÿ:™:×5Ñ5Ó7ˆˆZÒÙ	Ü Ÿ9™9×4Ñ4Ó6ˆˆZÒÙ	Ü Ÿ9™9×4Ñ4Ó6ˆˆZÒÙ	Ü!ŸJ™J×6Ñ6Ó8ˆˆ[ÒÙ	Ü!ŸJ™J×6Ñ6Ó8ˆˆ[ÒÙ	Ü#Ÿl™l×:Ñ:Ó<ˆˆ]ÒÙ	Ü$Ÿ}™}×1Ñ1ˆˆ^Ñä	Ð
BÔCÜ	ˆ$�)‰)¸¿¹¼ÔF¹©I¨D°#�r˜$˜˜r # Ò'¸ÒFÓ
GÔHä	¨d×.>Ñ.>Ð.FÑ
*ÐLkÔlô Ð'¬Ô.ð 	�	‰	Ð9J×9PÑ9PÔ9RÔSÑ9R©I¨D°#�T˜$˜˜r # Ò'Ð9RÒSÔTàÐ#Ð$Ð%ð ô
 
Ð
Ô à#4€DÐ	Ñ à€Kùó Gùó Ts   ËM,Ì;M2
Úreturnc                  óN   — t        «       } | j                  «       }t        |«       y)Nr   )r   Ú
parse_argsr   )r   rO   s     r   Úmainrc   ‡   s#   € ÜÓ!€FØ×ÑÓ€DÜ�ÔØr   Ú__main__)N)r   r3   r=   r8   Únumpyr?   r@   r/   r'   r   r<   Úaccelerate.commands.configr   r   Úutilsr   r   r	   r
   r   r   r   r   Úintrc   Ú__name__Ú
SystemExit© r   r   Ú<module>rl      s`   ðó" Û 	Û Û ã Û Û å -ß Q÷÷ óòNðbˆcó ð ˆzÒÙ
‘T“VÓ
Ðð r   