Correct flags

arcee-ai · Jan 3, 2024 · 1ef73cb · 1ef73cb
1 parent 18ecdf2
commit 1ef73cb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/mergekit/scripts/bakllama.py b/mergekit/scripts/bakllama.py
@@ -46,6 +46,7 @@ class BakllamaConfig(BaseModel):
 @click.option(
     "--clone-tensors/--no-clone-tensors",
     type=bool,
+    is_flag=True,
     help="Clone tensors before saving, to allow multiple occurrences of the same layer",
     default=False,
 )

diff --git a/mergekit/scripts/run_yaml.py b/mergekit/scripts/run_yaml.py
@@ -24,9 +24,11 @@
 
 
 @click.command("mergekit-yaml")
-@click.argument("config_file")  # , help="YAML configuration file")
-@click.argument("out_path")  # , help="Path to write result model")
-@click.option("--verbose", "-v", type=bool, default=False, help="Verbose logging")
+@click.argument("config_file")
+@click.argument("out_path")
+@click.option(
+    "--verbose", "-v", type=bool, default=False, is_flag=True, help="Verbose logging"
+)
 @add_merge_options
 def main(
     merge_options: MergeOptions,