We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
show_progress
1 parent a2675dc commit 6dd7636Copy full SHA for 6dd7636
src/llmcompressor/transformers/sparsification/compressed_tensors_utils.py
@@ -120,7 +120,9 @@ def save_pretrained_wrapper(
120
121
# make sure we're on the main process when saving
122
if state_dict is not None and len(state_dict) > 0:
123
- compressed_state_dict = compressor.compress(model, state_dict)
+ compressed_state_dict = compressor.compress(
124
+ model, state_dict, show_progress=True
125
+ )
126
logger.info("Saving compressed model to disk")
127
original_save_pretrained.__get__(model, model_class)(
128
save_directory,
0 commit comments