
    ^i#                    :    S r SSKJr  SSKrSSSS.r1 SkrS
S	 jrg)z<Argument parsing helpers shared by ACE-Step CLI entrypoints.    )annotationsNint8_weight_onlyfp8_weight_onlyw8a8_dynamic)r   r   r   >    nonenullc                    U c  gU R                  5       R                  5       nU[        ;   a  g[        R	                  U5      nUb  U$ [
        R                  " S5      e)zParse ``--quantization`` values from CLI input.

Args:
    value: Raw CLI value.

Returns:
    Canonical quantization method or ``None`` for disabled quantization.

Raises:
    argparse.ArgumentTypeError: If the value is not supported.
NzYInvalid quantization value. Use int8_weight_only, fp8_weight_only, w8a8_dynamic, or none.)striplower_NONE_ALIASES_QUANTIZATION_ALIASESgetargparseArgumentTypeError)value
normalizedquantizations      //mnt/workspace/ACE-Step-1.5/acestep/cli_args.pyparse_quantization_argr      s_     }$$&J]"(,,Z8L

$
$	!     )r   
str | Nonereturnr   )__doc__
__future__r   r   r   r   r    r   r   <module>r      s-    B "  +(" 
 %r   