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

Signed-off-by: Andy Xie <andy.xning@gmail.com>
2025-12-09 03:54:57 +08:00 · 2025-07-28 15:06:52 +08:00 · 2025-07-28 15:06:52 +08:00 · 150d9e6337
commit 150d9e6337
parent 139a97ec56
1 changed files with 1 additions and 1 deletions
--- 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",
    )