
    /h                         S SK r SSKJrJrJrJrJrJr  SSKJ	r	  / SQr
SS jr/ SS 4S jrS	 rS
 rS rS rS rS rS r " S S\ R(                  5      rg)    N   )ComputeEnvironmentDistributedTypeDynamoBackendFP8BackendTypePrecisionTypeSageMakerDistributedType   )
BulletMenu)EAGER	AOT_EAGERINDUCTORAOT_TS_NVFUSERNVPRIMS_NVFUSER
CUDAGRAPHSOFIFX2TRTONNXRTTENSORRTAOT_TORCHXLA_TRACE_ONCETORHCHXLA_TRACE_ONCEIPEXTVMc                     SnU(       a-  [        U 5      n Ub  [        U5      S:X  a  U$ Ub  U" U5      $ U$ g ! [         a    Ub  [        U5         Of = fU(       a  MU  g )NTr   )inputlen	Exceptionprint)
input_textconvert_valuedefaulterror_message	ask_againresults         a/home/james-whalen/.local/lib/python3.13/site-packages/accelerate/commands/config/config_utils.py
_ask_fieldr&   0   sp    I
z"	%"s6{a'7,9,E=(Q6Q   	%(m$	% )s   8 
8 8 AAc                 P    [        X5      nUR                  US9nUb  U" U5      $ U$ )N)default_choice)r   run)r   optionsr    r!   menur$   s         r%   _ask_optionsr,   =   s1    j*DXXWX-F$1$== I6I    c                 8    [        U 5      n [        SS/U    5      $ )NLOCAL_MACHINEAMAZON_SAGEMAKER)intr   values    r%   _convert_compute_environmentr4   C   s!    JE0BCEJKKr-   c                 8    [        U 5      n [        / SQU    5      $ )N)
NO	MULTI_CPU	MULTI_XPU	MULTI_HPU	MULTI_GPU	MULTI_NPU	MULTI_MLU
MULTI_SDAA
MULTI_MUSAXLA)r1   r   r2   s    r%   _convert_distributed_moder@   H   s'    JE	
 	 r-   c                 P    [        U 5      n [        [        U    5      R                  $ )N)r1   r   DYNAMO_BACKENDSr3   r2   s    r%   _convert_dynamo_backendrC   Z   s!    JE/0666r-   c                 8    [        U 5      n [        / SQU    5      $ )N)nofp16bf16fp8)r1   r   r2   s    r%   _convert_mixed_precisionrI   _   s    JE6u=>>r-   c                 8    [        U 5      n [        / SQU    5      $ )N)r6   DATA_PARALLELMODEL_PARALLEL)r1   r	   r2   s    r%   #_convert_sagemaker_distributed_moderM   d   s    JE#$Me$TUUr-   c                 8    [        U 5      n [        SS/U    5      $ )NTEMSAMP)r1   r   r2   s    r%   _convert_fp8_backendrQ   i   s    JE4/%011r-   c                 .    SSS.U R                  5          $ )NTF)yesrE   )lowerr2   s    r%   _convert_yes_no_to_boolrU   n   s    u%ekkm44r-   c                   ,   ^  \ rS rSrSrU 4S jrSrU =r$ )SubcommandHelpFormatterr   z[
A custom formatter that will remove the usage line from the help message for subcommands.
c                 L   > [         TU ]  XX45      nUR                  SS5      nU$ )Nz<command> [<args>]  )super_format_usagereplace)selfusageactionsgroupsprefix	__class__s        r%   r\   %SubcommandHelpFormatter._format_usagew   s*    %efE3R8r-    )__name__
__module____qualname____firstlineno____doc__r\   __static_attributes____classcell__)rc   s   @r%   rW   rW   r   s     r-   rW   )NNN)argparseutils.dataclassesr   r   r   r   r   r	   r+   r   rB   r&   r,   r4   r@   rC   rI   rM   rQ   rU   RawDescriptionHelpFormatterrW   re   r-   r%   <module>rp      sk   "   $
% &(tQ JL
$7
?
V
2
5hBB r-   