o
    d۷i                     @   sF   d dl mZ d dlmZmZ dgZejejjdG dd deZ	dS )    )Optional)CompressionFormatSparsityCompressionConfigBitmaskConfig)namec                   @   sB   e Zd ZU dZejjZee	d< dZ
ee e	d< dZee e	d< dS )r   z
    Configuration for storing a sparse model using
    bitmask compression

    :param global_sparsity: average sparsity of the entire model
    :param sparsity_structure: structure of the sparsity, such as
    "unstructured", "2:4", "8:16" etc
    formatg        global_sparsityunstructuredsparsity_structureN)__name__
__module____qualname____doc__r   sparse_bitmaskvaluer   str__annotations__r   r   floatr
    r   r   ^/home/ubuntu/vllm_env/lib/python3.10/site-packages/compressed_tensors/config/sparse_bitmask.pyr      s
   
 	N)
typingr   compressed_tensors.configr   r   __all__registerr   r   r   r   r   r   r   <module>   s
   