o
    Û·iŠ  ã                   @  s   d dl mZ ddd„ZdS )	é    )ÚannotationsÚquantize_typeú
str | NoneÚreturnc                 C  s&   | d u rd S dddddœ}|  | | ¡S )NÚfloat8_weight_onlyÚint8_weight_onlyÚint4_weight_onlyÚbitsandbytes_4bit)Ú	float8_woÚint8_woÚint4_woÚbnb_4bit)Úget)r   Úmapping© r   úN/home/ubuntu/vllm_env/lib/python3.10/site-packages/cache_dit/quantize/utils.pyÚnormalize_quantize_type   s   ür   N)r   r   r   r   )Ú
__future__r   r   r   r   r   r   Ú<module>   s    