feat: processors 支持批量 tokenize,优化性能并缓存 chat template

This commit is contained in:
2026-08-05 12:21:25 +08:00
parent aa2ea4f3a6
commit 65dadac10f
10 changed files with 290 additions and 25 deletions
+7
View File
@@ -53,6 +53,12 @@ def main():
parser.add_argument(
"--pad-value", type=int, default=0, help="Padding value (default: 0)"
)
parser.add_argument(
"--batch-size",
type=int,
default=256,
help="Records tokenized per batch (default: 256)",
)
parser.add_argument(
"--log-level",
default="INFO",
@@ -105,6 +111,7 @@ def main():
processor=processor,
pack_size=args.pack_size,
pad_value=args.pad_value,
batch_size=args.batch_size,
)
print(f"\nDone! Output saved to {output_dir}")