From de8de10aee1ab47f24b22da4c7cdca4f0b556d87 Mon Sep 17 00:00:00 2001 From: Irene Dea Date: Mon, 14 Oct 2024 18:36:22 -0700 Subject: [PATCH] Fix typo in eval code by using 'fsdp' instead of 'fsdp_config' (#1593) --- llmfoundry/command_utils/eval.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llmfoundry/command_utils/eval.py b/llmfoundry/command_utils/eval.py index 0678925a84..f25f2b5cef 100644 --- a/llmfoundry/command_utils/eval.py +++ b/llmfoundry/command_utils/eval.py @@ -110,7 +110,7 @@ def evaluate_model( mosaicml_logger._flush_metadata(force_flush=True) fsdp_config = parallelism_config.get( - 'fsdp_config', + 'fsdp', None, ) if parallelism_config else None if fsdp_config and model.get('load_in_8bit', False):