Update src/sparseml/transformers/compression/compressors/base.py

Co-authored-by: Michael Goin <[email protected]>

src/sparseml/transformers/compression/compressors/base.py

@@ -74,7 +74,7 @@ def overwrite_weights(self, model_path: str, model: Module):
         """
         Overwrites the weights in model with weights decompressed from model_path
 
-        :param pretrained_model_path: path to compressed weights
+        :param model_path: path to compressed weights
         :param model: pytorch model to load decompressed weights into
         """
         dense_gen = self.decompress(model_path)