From 124b60a5288b782cab5bb1f5314d528ef978dcb2 Mon Sep 17 00:00:00 2001 From: Max Marion Date: Mon, 12 Feb 2024 18:43:05 +0000 Subject: [PATCH] add default option --- llmfoundry/utils/builders.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llmfoundry/utils/builders.py b/llmfoundry/utils/builders.py index bf960603a2..391b792a1c 100644 --- a/llmfoundry/utils/builders.py +++ b/llmfoundry/utils/builders.py @@ -52,8 +52,8 @@ def build_evaluators( tokenizer: PreTrainedTokenizerBase, device_eval_batch_size: int, icl_seq_len: int, - fewshot_random_seed: Optional[int], icl_subset_num_batches: Optional[int], + fewshot_random_seed: Optional[int] = None, ) -> Tuple[List[Evaluator], List[str], Optional[EvalGauntlet]]: evaluators = []