fix load and save planner config logic (#1385)

mosaicml · Jul 23, 2024 · 0bed4ff · 0bed4ff
1 parent eb41a6e
commit 0bed4ff
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/llmfoundry/command_utils/train.py b/llmfoundry/command_utils/train.py
@@ -260,7 +260,7 @@ def train(cfg: DictConfig) -> Trainer:
 
     if fsdp_config is not None:
         if 'load_planner' in fsdp_config:
-            load_planners = fsdp_config['load_planner'].items()
+            load_planners = list(fsdp_config['load_planner'].items())
             if len(load_planners) > 1:
                 raise ValueError(
                     'Only one load planner can be specified in the config.',
@@ -272,7 +272,7 @@ def train(cfg: DictConfig) -> Trainer:
             )
 
         if 'save_planner' in fsdp_config:
-            save_planners = fsdp_config['save_planner'].items()
+            save_planners = list(fsdp_config['save_planner'].items())
             if len(save_planners) > 1:
                 raise ValueError(
                     'Only one save planner can be specified in the config.',