fix vit quantcfg init error. (#709)

ModelTC · Jan 23, 2025 · ef2201d · ef2201d
1 parent 5c36aa7
commit ef2201d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lightllm/models/vit/model.py b/lightllm/models/vit/model.py
@@ -107,7 +107,7 @@ def _init_weights(self):
         return
 
     def _init_quant(self):
-        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)
         logger.info(f"Initial quantization. " f"The default quantization method is {self.quant_cfg.quant_type}")
 
     def _init_infer_layer(self):