oops

mosaicml · Apr 9, 2024 · 142518a · 142518a
1 parent 710c9b0
commit 142518a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/train/train.py b/scripts/train/train.py
@@ -92,7 +92,7 @@ class TrainConfig:
     load_strict_model_weights: bool = True
     load_ignore_keys: Optional[List[str]] = None
     compile_config: Optional[Dict[str, Any]] = None
-    metadata: Optional[Dict[str, str]] = None
+    metadata: Optional[DictConfig] = None
     log_config: bool = True
     autoresume: bool = False
     data_local: Optional[Dict[str, Any]] = None