diff --git a/models/chatglm_llm.py b/models/chatglm_llm.py
index ec2c04e..c5e4738 100644
--- a/models/chatglm_llm.py
+++ b/models/chatglm_llm.py
@@ -141,16 +141,16 @@ class ChatGLM(LLM):
             else:
                 from accelerate import dispatch_model
 
-                model = AutoModel.from_pretrained(model_name_or_path, trust_remote_code=True,
-                                                  config=model_config, **kwargs)
+                # model = AutoModel.from_pretrained(model_name_or_path, trust_remote_code=True,
+                #         config=model_config, **kwargs)
                 if LLM_LORA_PATH and use_lora:
                     from peft import PeftModel
-                    model = PeftModel.from_pretrained(model, LLM_LORA_PATH)
+                    model = PeftModel.from_pretrained(self.model, LLM_LORA_PATH)
                 # 可传入device_map自定义每张卡的部署情况
                 if device_map is None:
                     device_map = auto_configure_device_map(num_gpus)
 
-                self.model = dispatch_model(model.half(), device_map=device_map)
+                self.model = dispatch_model(self.model.half(), device_map=device_map)
         else:
             self.model = self.model.float().to(llm_device)