fix

mosaicml · Feb 14, 2024 · 2c50446 · 2c50446
1 parent 5a47002
commit 2c50446
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/llmfoundry/utils/builders.py b/llmfoundry/utils/builders.py
@@ -555,8 +555,7 @@ def _validate_cfg(icl_cfg: DictConfig):
                 cot_delimiter=icl_cfg.get('cot_delimiter', ''),
                 generation_kwargs=icl_cfg.get('generation_kwargs', {}),
                 early_stopping_criteria=early_stopping_criteria,
-                do_normalization=icl_cfg.get('do_normalization', True),
-                generation_kwargs=icl_cfg.get('generation_kwargs'))
+                do_normalization=icl_cfg.get('do_normalization', True))
             if hasattr(
                     icl_cfg,
                     'has_categories') and icl_cfg.has_categories and isinstance(