Use default composer run name instead of llm

mosaicml · Oct 23, 2024 · c681a9c · c681a9c
1 parent 21c7ec8
commit c681a9c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/llmfoundry/command_utils/train.py b/llmfoundry/command_utils/train.py
@@ -312,9 +312,9 @@ def train(cfg: DictConfig) -> Trainer:
 
     # Optional parameters will be set to default values if not specified.
     env_run_name: Optional[str] = os.environ.get('RUN_NAME', None)
-    run_name: str = (
+    run_name: Optional[str] = (
         train_cfg.run_name if train_cfg.run_name else env_run_name
-    ) or 'llm'
+    )
     is_state_dict_sharded: bool = (
         fsdp_config.get('state_dict_type', 'full') == 'sharded'
     ) if fsdp_config else False