Skip to content

Commit 1cce626

Browse files
authored
peft: slightly improved snippet? (#644)
1 parent 779c9f1 commit 1cce626

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

packages/tasks/src/model-libraries-snippets.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -496,8 +496,8 @@ export const peft = (model: ModelData): string[] => {
496496
from transformers import AutoModelFor${pefttask}
497497
498498
config = PeftConfig.from_pretrained("${model.id}")
499-
model = AutoModelFor${pefttask}.from_pretrained("${peftBaseModel}")
500-
model = PeftModel.from_pretrained(model, "${model.id}")`,
499+
base_model = AutoModelFor${pefttask}.from_pretrained("${peftBaseModel}")
500+
model = PeftModel.from_pretrained(base_model, "${model.id}")`,
501501
];
502502
};
503503

0 commit comments

Comments
 (0)