Compare commits

...

2 Commits

Author SHA1 Message Date
f0bb5f0ed5 Fix default config dicts (#329)
* Fix default config dicts

* style
2022-04-28 11:24:24 -04:00
11e8b33217 Release: v0.7.0 2022-04-28 11:03:50 -04:00
3 changed files with 9 additions and 2 deletions

View File

@ -36,7 +36,7 @@ extras["sagemaker"] = [
setup(
name="accelerate",
version="0.7.0.dev0",
version="0.7.0",
description="Accelerate",
long_description=open("README.md", "r", encoding="utf-8").read(),
long_description_content_type="text/markdown",

View File

@ -2,7 +2,7 @@
# There's no way to ignore "F401 '...' imported but unused" warnings in this
# module, but to preserve other warnings. So, don't check this module at all.
__version__ = "0.7.0.dev0"
__version__ = "0.7.0"
from .accelerator import Accelerator
from .kwargs_handlers import DistributedDataParallelKwargs, GradScalerKwargs, InitProcessGroupKwargs

View File

@ -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):