Fix typo

fishaudio · Dec 29, 2023 · e486010 · e486010
1 parent 74ec1db
commit e486010
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/llama/build_dataset.py b/tools/llama/build_dataset.py
@@ -122,7 +122,7 @@ def run_task(task):
 )
 @click.option("--output", type=click.Path(), default="data/quantized-dataset-ft.protos")
 @click.option("--filelist", type=click.Path(), default=None)
-@click.option("--num_workers", type=int, default=16)
+@click.option("--num-workers", type=int, default=16)
 def main(config, output, filelist, num_workers):
     dataset_fp = open(output, "wb")
     generator_fn = (