
    *`iy                         d dl mZ d dlmZmZmZ dgZ ej        ej        j	                   G d de                      Z
dS )    )Optional)CompressionFormatSparsityCompressionConfigSparsityStructureSparse24BitMaskConfig)namec                   ~    e Zd ZU dZej        j        Zee	d<   dZ
ee         e	d<   ej        j        Zee         e	d<   dS )r   a  
    Configuration for storing a 24 sparse model using
    bytemask compression

    :param global_sparsity: average sparsity of the entire model
    :param sparsity_structure: structure of the sparsity, should always be
        "2:4" for this compression format
    formatg        global_sparsitysparsity_structureN)__name__
__module____qualname____doc__r   sparse_24_bitmaskvaluer
   str__annotations__r   r   floatr   TWO_FOURr        /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/compressed_tensors/config/sparse_24_bitmask.pyr   r      sc           $5;FC;;;'*OXe_***(9(B(HHHHHHr   N)typingr   compressed_tensors.configr   r   r   __all__registerr   r   r   r   r   r   <module>r      s                   #
# $#):)L)RSSSI I I I I5 I I TSI I Ir   