Skip to content

Commit 0c6f1d5

Browse files
committed
fix: quality check
Signed-off-by: Mehant Kammakomati <mehant.kammakomati2@ibm.com>
1 parent 92fd856 commit 0c6f1d5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/accelerate/utils/fsdp_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -630,7 +630,7 @@ def fsdp2_prepare_model(accelerator, model: torch.nn.Module) -> torch.nn.Module:
630630
# `fully_shard` doesn't accept `None` in case of `MixedPrecisionPolicy`
631631
"mp_policy": fsdp2_plugin.mixed_precision_policy or MixedPrecisionPolicy(),
632632
"mesh": mesh[tuple(accelerator.parallelism_config.fsdp_dim_names)] if mesh is not None else None,
633-
"ignored_params": get_parameters_from_modules(fsdp2_plugin.ignored_modules, model, accelerator.device)
633+
"ignored_params": get_parameters_from_modules(fsdp2_plugin.ignored_modules, model, accelerator.device),
634634
}
635635

636636
model_has_params4bit = False

0 commit comments

Comments
 (0)