Skip to content

Commit f0bb5f0

Browse files
committed
Fix default config dicts (#329)
* Fix default config dicts * style
1 parent 11e8b33 commit f0bb5f0

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

src/accelerate/commands/config/config_args.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -139,6 +139,13 @@ class ClusterConfig(BaseConfig):
139139
# args for fsdp
140140
fsdp_config: dict = None
141141

142+
def __post_init__(self):
143+
if self.deepspeed_config is None:
144+
self.deepspeed_config = {}
145+
if self.fsdp_config is None:
146+
self.fsdp_config = {}
147+
return super().__post_init__()
148+
142149

143150
@dataclass
144151
class SageMakerConfig(BaseConfig):

0 commit comments

Comments
 (0)