efort
/
glm4
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add del label

This commit is contained in:
zR 2024-12-15 12:43:53 +08:00
parent 52cd14ce8c
commit 81ba7e087c
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
finetune_demo/finetune_vision.py
View File

@ -77,6 +77,7 @@ class Seq2SeqTrainer(_Seq2SeqTrainer):
with torch.no_grad(): with torch.no_grad():
if self.args.predict_with_generate: if self.args.predict_with_generate:
output_ids = inputs.pop("output_ids", None) output_ids = inputs.pop("output_ids", None)
del inputs["labels"]
loss, generated_tokens, labels = super().prediction_step( loss, generated_tokens, labels = super().prediction_step(
model=model, model=model,
inputs=inputs, inputs=inputs,