Fix default config dicts (#329)

* Fix default config dicts * style
huggingface · Apr 28, 2022 · f0bb5f0 · f0bb5f0
1 parent 11e8b33
commit f0bb5f0
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/accelerate/commands/config/config_args.py b/src/accelerate/commands/config/config_args.py
@@ -139,6 +139,13 @@ class ClusterConfig(BaseConfig):
     # args for fsdp
     fsdp_config: dict = None
 
+    def __post_init__(self):
+        if self.deepspeed_config is None:
+            self.deepspeed_config = {}
+        if self.fsdp_config is None:
+            self.fsdp_config = {}
+        return super().__post_init__()
+
 
 @dataclass
 class SageMakerConfig(BaseConfig):