Unverified Commit c45aab75 authored by Martin Malmsten's avatar Martin Malmsten Committed by GitHub
Browse files

Added missing parenthesis in call to is_fsdp_enabled (#25585)

Calling function is_fsdp_enabled instead of checking if it is not None
parent 940d1a76
......@@ -468,7 +468,7 @@ def load_state_dict(checkpoint_file: Union[str, os.PathLike]):
return safe_load_file(checkpoint_file)
try:
if (
(is_deepspeed_zero3_enabled() or is_fsdp_enabled)
(is_deepspeed_zero3_enabled() or is_fsdp_enabled())
and torch.distributed.is_initialized()
and torch.distributed.get_rank() > 0
):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment