feature: adds --control-strength as parameter for cli

brycedrennan · Dec 6, 2023 · ef0f446 · ef0f446
1 parent 24f4af3
commit ef0f446
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/imaginairy/cli/imagine.py b/imaginairy/cli/imagine.py
@@ -32,6 +32,13 @@
     ),
     multiple=True,
 )
+@click.option(
+    "--control-strength",
+    help=(
+        "Strength of the control signal."
+    ),
+    multiple=True,
+)
 @click.option(
     "--control-mode",
     default=None,
@@ -106,6 +113,7 @@ def imagine_cmd(
     caption_text,
     control_image,
     control_image_raw,
+    control_strength,
     control_mode,
     videogen,
 ):
@@ -146,6 +154,7 @@ def imagine_cmd(
                 ControlNetInput(
                     image=control_image,
                     image_raw=control_image_raw,
+                    strength=float(control_strength[i]),
                     mode=cm,
                 )
             )