o iy@sJddlmZddlmZmZmZdgZejejj dGdddeZ dS))Optional)CompressionFormatSparsityCompressionConfigSparsityStructureSparse24BitMaskConfig)namec@sFeZdZUdZejjZee d<dZ e e e d<e jjZe ee d<dS)ra  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 formatgglobal_sparsitysparsity_structureN)__name__ __module__ __qualname____doc__rsparse_24_bitmaskvaluerstr__annotations__r rfloatrTWO_FOURr rr_/home/ubuntu/.local/lib/python3.10/site-packages/compressed_tensors/config/sparse_24_bitmask.pyrs  N) typingrcompressed_tensors.configrrr__all__registerrrrrrrrs