File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
src/llmcompressor/transformers/sparsification Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -209,19 +209,19 @@ def get_model_compressor(
209209 if state_dict is None :
210210 state_dict = get_state_dict_offloaded_model (model )
211211
212- sparsity_stucture = SparsityConfigMetadata .infer_sparsity_structure (model )
212+ sparsity_structure = SparsityConfigMetadata .infer_sparsity_structure (model )
213213 quantization_format : Optional [CompressionFormat ] = infer_quantization_format (
214214 model = model ,
215215 quantization_format = quantization_format ,
216216 save_compressed = save_compressed ,
217- sparsity_structure = sparsity_stucture ,
217+ sparsity_structure = sparsity_structure ,
218218 )
219219
220220 if sparsity_config is not None :
221221 sparsity_config .global_sparsity = SparsityConfigMetadata .infer_global_sparsity (
222222 model , state_dict = state_dict
223223 )
224- sparsity_config .sparsity_structure = sparsity_stucture
224+ sparsity_config .sparsity_structure = sparsity_structure
225225 elif not skip_compression_stats :
226226 # try to infer a sparsity config from the model if none is provided
227227 logger .info (
You can’t perform that action at this time.
0 commit comments