Update cli.py

t5s/cli.py

@@ -35,6 +35,9 @@ parser_start.add_argument(
 )
 parser_start.add_argument(
     "-m",
+    "--model_name",
+    default="t5-base",
+    help="Enter the model to be used eg t5-base",
 parser_dirs = command_subparser.add_parser(
     "dirs",
     help="Create directories that are ignored by git but required for " "the project",