machineuser commited on
Commit
c25d7cc
1 Parent(s): f386dcf

Sync widgets demo

Browse files
packages/tasks/src/model-libraries-snippets.ts CHANGED
@@ -496,8 +496,8 @@ export const peft = (model: ModelData): string[] => {
496
  from transformers import AutoModelFor${pefttask}
497
 
498
  config = PeftConfig.from_pretrained("${model.id}")
499
- model = AutoModelFor${pefttask}.from_pretrained("${peftBaseModel}")
500
- model = PeftModel.from_pretrained(model, "${model.id}")`,
501
  ];
502
  };
503
 
 
496
  from transformers import AutoModelFor${pefttask}
497
 
498
  config = PeftConfig.from_pretrained("${model.id}")
499
+ base_model = AutoModelFor${pefttask}.from_pretrained("${peftBaseModel}")
500
+ model = PeftModel.from_pretrained(base_model, "${model.id}")`,
501
  ];
502
  };
503