From ef0f44646e2dfb0b6e0ffc8d77c005410d2e974c Mon Sep 17 00:00:00 2001
From: jaydrennan <jsdman1313@gmail.com>
Date: Tue, 5 Dec 2023 18:39:56 -0800
Subject: [PATCH] feature: adds --control-strength as parameter for cli

---
 imaginairy/cli/imagine.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/imaginairy/cli/imagine.py b/imaginairy/cli/imagine.py
index 96ed05ca..3ab8fb8a 100644
--- 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,
                 )
             )