Skip to content

Commit 6dd7636

Browse files
authored
Set show_progress to True (#1471)
Summary - Until we're using compress_model, we need to show compression progress - Without it, it seems like nothing is happening
1 parent a2675dc commit 6dd7636

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/llmcompressor/transformers/sparsification/compressed_tensors_utils.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,9 @@ def save_pretrained_wrapper(
120120

121121
# make sure we're on the main process when saving
122122
if state_dict is not None and len(state_dict) > 0:
123-
compressed_state_dict = compressor.compress(model, state_dict)
123+
compressed_state_dict = compressor.compress(
124+
model, state_dict, show_progress=True
125+
)
124126
logger.info("Saving compressed model to disk")
125127
original_save_pretrained.__get__(model, model_class)(
126128
save_directory,

0 commit comments

Comments
 (0)