change the default value of fsdp_min_num_params to int (#3902)

CodeMan62 · web-flow · commit c7a774605c23 · 2026-01-07T15:51:29.000+01:00
diff --git a/src/accelerate/commands/launch.py b/src/accelerate/commands/launch.py
@@ -526,7 +526,7 @@ def launch_command_parser(subparsers=None):
     fsdp_args.add_argument(
         "--fsdp_min_num_params",
         type=int,
-        default=1e8,
+        default=int(1e8),
         help="FSDP's minimum number of parameters for Default Auto Wrapping. (useful only when `use_fsdp` flag is passed).",
     )
     # We enable this for backwards compatibility, throw a warning if this is set in `FullyShardedDataParallelPlugin`

Original file line number	Diff line number	Diff line change
`@@ -526,7 +526,7 @@ def launch_command_parser(subparsers=None):`
`526`	`526`	`fsdp_args.add_argument(`
`527`	`527`	`"--fsdp_min_num_params",`
`528`	`528`	`type=int,`
`529`		`- default=1e8,`
	`529`	`+ default=int(1e8),`
`530`	`530`	help="FSDP's minimum number of parameters for Default Auto Wrapping. (useful only when `use_fsdp` flag is passed).",
`531`	`531`	`)`
`532`	`532`	# We enable this for backwards compatibility, throw a warning if this is set in `FullyShardedDataParallelPlugin`