"tests/vscode:/vscode.git/clone" did not exist on "3668ec17165dbb7823f3bc7e190e1733040c3af8"
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]): ...@@ -468,7 +468,7 @@ def load_state_dict(checkpoint_file: Union[str, os.PathLike]):
return safe_load_file(checkpoint_file) return safe_load_file(checkpoint_file)
try: try:
if ( 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.is_initialized()
and torch.distributed.get_rank() > 0 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