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.
1 parent 5c36aa7 commit ef2201dCopy full SHA for ef2201d
lightllm/models/vit/model.py
@@ -107,7 +107,7 @@ def _init_weights(self):
107
return
108
109
def _init_quant(self):
110
- self.quant_cfg = Quantcfg(self.config["num_hidden_layers"], self.quant_type, self.quant_cfg_path)
+ self.quant_cfg = Quantcfg(self.config, self.quant_type, self.quant_cfg_path)
111
logger.info(f"Initial quantization. " f"The default quantization method is {self.quant_cfg.quant_type}")
112
113
def _init_infer_layer(self):
0 commit comments