Skip to content

Commit 7282c24

Browse files
authored
Update sft_trainer.py
Signed-off-by: Greenewald <greenewk@umich.edu>
1 parent be61881 commit 7282c24

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tuning/sft_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -382,7 +382,7 @@ def train(
382382
tokenizer.padding_side='right'
383383
response_token_ids = (tokenizer(peft_config.invocation_string, return_tensors="pt", add_special_tokens=False))['input_ids']
384384
model = aLoRAPeftModelForCausalLM(model, peft_config, response_token_ids = response_token_ids)
385-
data_collator = DataCollatorForCompletionOnlyLM(invocation_string, tokenizer=tokenizer)
385+
386386
peft_config = None
387387
trainer = SFTTrainer(
388388
model=model,

0 commit comments

Comments
 (0)