From 0d7300e5ff0a0a0f8e21b0122e1538501312cc82 Mon Sep 17 00:00:00 2001 From: Max Ryabinin Date: Fri, 19 Apr 2024 17:09:21 +0200 Subject: [PATCH] Replace underscores with dashes in argument names --- examples/tokenize_data.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/tokenize_data.py b/examples/tokenize_data.py index daaddf9..b03a1a2 100644 --- a/examples/tokenize_data.py +++ b/examples/tokenize_data.py @@ -95,7 +95,7 @@ def process_data(args: argparse.Namespace) -> None: "--max-seq-length", type=int, default=2048, help="Maximum sequence length" ) parser.add_argument( - "--add_labels", + "--add-labels", action="store_true", help="Whether to add loss labels from padding tokens", ) @@ -106,7 +106,7 @@ def process_data(args: argparse.Namespace) -> None: help="Tokenizer name (for example, togethercomputer/LLaMA-2-7B-32K)", ) parser.add_argument( - "--out_filename", + "--out-filename", default="processed_dataset.parquet", help="Name of the Parquet file to save (should have .parquet extension)", )