fix

mosaicml · Feb 2, 2024 · 40acdb4 · 40acdb4
1 parent c6f2109
commit 40acdb4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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,