mirror of
https://github.com/huggingface/peft.git
synced 2025-10-20 15:33:48 +08:00
FIX Deprecations in MiSS example (#2704)
Also, was validated on XPU. --------- Signed-off-by: Yao, Matrix <matrix.yao@intel.com>
This commit is contained in:
@ -41,7 +41,7 @@ trainer = SFTTrainer(
|
||||
model=peft_model,
|
||||
args=training_args,
|
||||
train_dataset=dataset,
|
||||
tokenizer=tokenizer,
|
||||
processing_class=tokenizer,
|
||||
)
|
||||
trainer.train()
|
||||
peft_model.save_pretrained("miss-llama-2-7b")
|
||||
@ -101,4 +101,4 @@ python miss_finetuning.py \
|
||||
archivePrefix={arXiv},
|
||||
primaryClass={cs.CL},
|
||||
url={https://arxiv.org/abs/2409.15371},
|
||||
}
|
||||
}
|
||||
|
@ -92,7 +92,7 @@ trainer = SFTTrainer(
|
||||
model=peft_model,
|
||||
args=script_args,
|
||||
train_dataset=dataset,
|
||||
tokenizer=tokenizer,
|
||||
processing_class=tokenizer,
|
||||
)
|
||||
trainer.train()
|
||||
trainer.save_state()
|
||||
|
Reference in New Issue
Block a user