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:
Yao Matrix
2025-08-05 02:46:28 -07:00
committed by GitHub
parent ec7dee024f
commit 2518ceeb15
2 changed files with 3 additions and 3 deletions

View File

@ -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},
}
}

View File

@ -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()