diff --git a/skynet/cli.py b/skynet/cli.py
index e27d9da..5415ecf 100755
--- a/skynet/cli.py
+++ b/skynet/cli.py
@@ -106,7 +106,7 @@ def inpaint(model, prompt, input, mask, output, strength, guidance, steps, seed)
 @click.option('--temperature', '-t', default=1.0)
 @click.option('--max-length', '-ml', default=256)
 @click.option('--num-return-sequences', '-rs', default=1)
-@click.option('--no-repeat-ngram', '-nr', default=2)
+@click.option('--no-repeat-ngram_size', '-nr', default=2)
 @click.option('--top-p', '-tp', default=0.95)
 def txt2txt(
     model,
@@ -115,7 +115,7 @@ def txt2txt(
     temperature,
     max_length,
     num_return_sequences,
-    no_repeat_ngram,
+    no_repeat_ngram_size,
     top_p
 ):
     from . import utils
@@ -131,7 +131,7 @@ def txt2txt(
         temperature=temperature,
         max_length=max_length,
         num_return_sequences=num_return_sequences,
-        no_repeat_ngram=no_repeat_ngram,
+        no_repeat_ngram_size=no_repeat_ngram_size,
         top_p=top_p
     )