fix test

mosaicml · Jun 6, 2024 · 1d8d104 · 1d8d104
1 parent ef3f657
commit 1d8d104
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/llmfoundry/utils/config_utils.py b/llmfoundry/utils/config_utils.py
@@ -76,6 +76,9 @@ class EvalConfig:
     run_name: Optional[str] = None
     metadata: Optional[Dict[str, str]] = None
 
+    # Dataloader
+    global_train_batch_size: Optional[int] = None
+
     # Distributed parameters
     dist_timeout: Union[float, int] = 600.0
     fsdp_config: Optional[Dict[str, Any]] = None