From 2a4173789f17fbc53943191db92dae5be6780ec4 Mon Sep 17 00:00:00 2001 From: Sara Adkins Date: Wed, 3 Apr 2024 10:11:29 -0400 Subject: [PATCH] Update src/sparseml/transformers/compression/compressors/base.py Co-authored-by: Michael Goin --- src/sparseml/transformers/compression/compressors/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/sparseml/transformers/compression/compressors/base.py b/src/sparseml/transformers/compression/compressors/base.py index 22166e4a7b3..2a1a37d9196 100644 --- a/src/sparseml/transformers/compression/compressors/base.py +++ b/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)