Selaa lähdekoodia

Update the --text-unk-blocking argument to be the same style as others (#221)

Co-authored-by: Peng-Jen Chen <pipibjc@devfair0728.h2.fair>
pipibjc 1 vuosi sitten
vanhempi
commit
b7b54d7e38
1 muutettua tiedostoa jossa 1 lisäystä ja 1 poistoa
  1. 1 1
      src/seamless_communication/cli/m4t/predict/predict.py

+ 1 - 1
src/seamless_communication/cli/m4t/predict/predict.py

@@ -136,7 +136,7 @@ def add_inference_arguments(parser: argparse.ArgumentParser) -> argparse.Argumen
         default=False,
     )
     parser.add_argument(
-        "--text-unk-blocking",
+        "--text_unk_blocking",
         type=bool,
         help=(
             "If True, set penalty of UNK to inf in text generator "