mask_history args verify valid

Former-commit-id: 2f8388b4f4195d934400ad9267d72e10ca4105a3
This commit is contained in:
“Wzw”
2024-08-08 10:12:01 +08:00
parent d71446e387
commit 6ec64a7e56
2 changed files with 5 additions and 0 deletions

View File

@@ -141,3 +141,6 @@ class DataArguments:
if self.streaming and self.max_samples is not None:
raise ValueError("`max_samples` is incompatible with `streaming`.")
if self.mask_history and self.train_on_prompt:
raise ValueError("`Train on the last turn only` does not support `train_on_prompt`.")