comments

neuralmagic · Jun 17, 2024 · d39e7f9 · d39e7f9
1 parent eb6ad2b
commit d39e7f9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/src/sparseml/modifiers/quantization/gptq/utils/gptq_wrapper.py b/src/sparseml/modifiers/quantization/gptq/utils/gptq_wrapper.py
@@ -89,6 +89,7 @@ def fasterprune(
         Run pruning and quantization(if applicable) on the layer up to the target
         sparsity value.
 
+        :param actorder: Flag to apply activation reordering
         :param blocksize: Number of columns to compress in one pass
         :param percdamp: Amount of dampening to apply to H, as a fraction of the
             diagonal norm

diff --git a/src/sparseml/modifiers/utils/layer_compressor.py b/src/sparseml/modifiers/utils/layer_compressor.py
@@ -134,6 +134,8 @@ def revert_layer_wrappers(self):
     def compress(self, actorder: bool = False):
         """
         Apply compression to each wrapped submodule in the layer
+        
+        :param: actorder: flag to apply activation reordering
         """
 
         @torch.no_grad()