Fix trainer.py not showing signature columns (#38465)

Fix trainer.py not showing signature columns
This commit is contained in:
Kusanagi Nene 2025-06-13 23:39:29 +08:00 committed by GitHub
parent 4c3c177ecf
commit 2ff964bcb4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -932,7 +932,7 @@ class Trainer:
columns = [k for k in signature_columns if k in dataset.column_names]
if len(columns) == 0:
raise ValueError(
"No columns in the dataset match the model's forward method signature: ({', '.join(signature_columns)}). "
f"No columns in the dataset match the model's forward method signature: ({', '.join(signature_columns)}). "
f"The following columns have been ignored: [{', '.join(ignored_columns)}]. "
"Please check the dataset and model. You may need to set `remove_unused_columns=False` in `TrainingArguments`."
)