Open jinlong7790 opened 5 months ago
model = LlamaForCausalLM.from_pretrained( base_model, load_in_8bit=load_8bit, torch_dtype=torch.float16, device_map="auto", trust_remote_code=True, ) model = PeftModel.from_pretrained( model, lora_weights, torch_dtype=torch.float16, ) model = PeftModel.from_pretrained(model, "/home/zzu_zxw/zjl_data/KnowPAT/save4")
Hi, 只需要加载一次就可以了,这里的第二次加载的代码是在哪个文件呢?