tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

make sure to define all the explicit positional args

Browse Source
This commit is contained in:
Wing Lian
2023-08-19 21:20:10 -04:00
parent 8d3c8a3eab
commit e3b07402a7
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/utils/callbacks.py
View File

@@ -152,6 +152,8 @@ def mmlu_eval_callback_factory(trainer, tokenizer):
def on_evaluate( def on_evaluate(
self, self,
args: AxolotlTrainingArguments, args: AxolotlTrainingArguments,
state: TrainerState, # pylint: disable=unused-argument
control: TrainerControl, # pylint: disable=unused-argument
**kwargs, # pylint: disable=unused-argument **kwargs, # pylint: disable=unused-argument
): ):
data_loader = trainer.get_eval_dataloader(mmlu_dataset) data_loader = trainer.get_eval_dataloader(mmlu_dataset)