diff --git a/packages/tasks/src/model-libraries-snippets.ts b/packages/tasks/src/model-libraries-snippets.ts index bb6ac12c3..3dc520e0c 100644 --- a/packages/tasks/src/model-libraries-snippets.ts +++ b/packages/tasks/src/model-libraries-snippets.ts @@ -933,10 +933,9 @@ export const peft = (model: ModelData): string[] => { } return [ - `from peft import PeftModel, PeftConfig + `from peft import PeftModel from transformers import AutoModelFor${pefttask} -config = PeftConfig.from_pretrained("${model.id}") base_model = AutoModelFor${pefttask}.from_pretrained("${peftBaseModel}") model = PeftModel.from_pretrained(base_model, "${model.id}")`, ];