[Bugfix] fix max-file-size type from str to int (vllm-project#21675)

andyxning · epwalsh · commit d3096168642f · 2025-08-27T16:55:29.000-07:00
Signed-off-by: Andy Xie &lt;andy.xning@gmail.com&gt;
diff --git a/examples/offline_inference/save_sharded_state.py b/examples/offline_inference/save_sharded_state.py
@@ -47,7 +47,7 @@ def parse_args():
     )
     parser.add_argument(
         "--max-file-size",
-        type=str,
+        type=int,
         default=5 * 1024**3,
         help="max size (in bytes) of each safetensors file",
     )

Original file line number	Diff line number	Diff line change
`@@ -47,7 +47,7 @@ def parse_args():`
`47`	`47`	`)`
`48`	`48`	`parser.add_argument(`
`49`	`49`	`"--max-file-size",`
`50`		`- type=str,`
	`50`	`+ type=int,`
`51`	`51`	`default=5 * 1024**3,`
`52`	`52`	`help="max size (in bytes) of each safetensors file",`
`53`	`53`	`)`