Skip to content
Snippets Groups Projects
Unverified Commit a6208317 authored by Huang Zhihong's avatar Huang Zhihong Committed by GitHub
Browse files

Fix the bug when continue the peft. (#717)

parent 9a56c419
Branches
No related tags found
No related merge requests found
......@@ -167,7 +167,7 @@ def main(**kwargs):
# Load the pre-trained peft model checkpoint and setup its configuration
if train_config.from_peft_checkpoint:
model = PeftModel.from_pretrained(model, train_config.from_peft_checkpoint, is_trainable=True)
peft_config = model.peft_config()
peft_config = model.peft_config
# Generate the peft config and start fine-tuning from original model
else:
peft_config = generate_peft_config(train_config, kwargs)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment