From 40acdb438390dc08fbd34320027cfe5e6c88f6e7 Mon Sep 17 00:00:00 2001 From: Jeremy Dohmann Date: Fri, 2 Feb 2024 15:44:34 -0500 Subject: [PATCH] fix --- llmfoundry/utils/builders.py | 1 + 1 file changed, 1 insertion(+) diff --git a/llmfoundry/utils/builders.py b/llmfoundry/utils/builders.py index eebd2ed7fa..76dbfb7883 100644 --- a/llmfoundry/utils/builders.py +++ b/llmfoundry/utils/builders.py @@ -509,6 +509,7 @@ def _validate_cfg(icl_cfg: DictConfig): None) if isinstance(early_stopping_criteria, ListConfig): early_stopping_criteria = om.to_container(early_stopping_criteria) + assert early_stopping_criteria is None or isinstance(early_stopping_criteria, list) dataloaders = get_icl_task_dataloader( icl_cfg.icl_task_type, icl_cfg.dataset_uri,