File tree Expand file tree Collapse file tree 1 file changed +2
-1
lines changed
src/compressed_tensors/compressors Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Original file line number Diff line number Diff line change @@ -81,6 +81,7 @@ class ModelCompressor:
81
81
def from_pretrained (
82
82
cls ,
83
83
pretrained_model_name_or_path : str ,
84
+ ** kwargs ,
84
85
) -> Optional ["ModelCompressor" ]:
85
86
"""
86
87
Given a path to a model config, extract the sparsity and/or quantization
@@ -89,7 +90,7 @@ def from_pretrained(
89
90
:param pretrained_model_name_or_path: path to model config on disk or HF hub
90
91
:return: compressor for the extracted configs
91
92
"""
92
- config = AutoConfig .from_pretrained (pretrained_model_name_or_path )
93
+ config = AutoConfig .from_pretrained (pretrained_model_name_or_path , ** kwargs )
93
94
compression_config = getattr (config , COMPRESSION_CONFIG_NAME , None )
94
95
return cls .from_compression_config (compression_config )
95
96
You can’t perform that action at this time.
0 commit comments