diff --git a/CGE/utils/vllm_codefuse_cge_large.py b/CGE/utils/vllm_codefuse_cge_large.py index 49c58e0..ab96331 100644 --- a/CGE/utils/vllm_codefuse_cge_large.py +++ b/CGE/utils/vllm_codefuse_cge_large.py @@ -496,7 +496,12 @@ def __init__( self.config = config self.lora_config = lora_config self.quant_config = quant_config - self.plm_model = Qwen2ForCausalLM(config, cache_config, quant_config) + self.plm_model = Qwen2ForCausalLM( + config=config, + cache_config=cache_config, + quant_config=quant_config, + lora_config=lora_config # 添加lora_config + ) self.embedding_method = config.embedding_method self.inf_seq_length = config.inf_seq_length self.padding_side = config.padding_side