diff --git a/src/llama_recipes/configs/wandb.py b/src/llama_recipes/configs/wandb.py index 6a43ffec2..24d137dec 100644 --- a/src/llama_recipes/configs/wandb.py +++ b/src/llama_recipes/configs/wandb.py @@ -12,4 +12,5 @@ class wandb_config: tags: Optional[List[str]] = None group: Optional[str] = None notes: Optional[str] = None - mode: Optional[str] = None \ No newline at end of file + mode: Optional[str] = None + name: Optional[str] = None