Skip to content

Commit

Permalink
Lint
Browse files Browse the repository at this point in the history
Signed-off-by: Rahul Tuli <[email protected]>
  • Loading branch information
rahul-tuli committed Jan 22, 2025
1 parent 0de4042 commit 96f376e
Showing 1 changed file with 2 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -264,7 +264,8 @@ def _decompress_bitmask_compressed_weight(
:param compressed: The 2:4 sparse weight tensor
compressed using the sparse-24-bitmask compressor.
:param bitmask: The 2:4 bitmask associated with the compressed weights.
:param layer: The layer whose weights need to be processed after loading.
:param layer: The layer whose weights need to be processed
after loading.
:return: The decompressed 2:4 sparse weight tensor.
"""

Expand Down

0 comments on commit 96f376e

Please sign in to comment.