From fa9a4f4a8e8079ab0b044550b39783b86fba1511 Mon Sep 17 00:00:00 2001
From: sudoskys <coldlando@hotmail.com>
Date: Sat, 24 Feb 2024 19:31:28 +0800
Subject: [PATCH] fix(sdk): Add reference_image reference_strength etc.

---
 src/novelai_python/sdk/ai/generate_image/__init__.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/novelai_python/sdk/ai/generate_image/__init__.py b/src/novelai_python/sdk/ai/generate_image/__init__.py
index a8269fd..48bc24a 100755
--- a/src/novelai_python/sdk/ai/generate_image/__init__.py
+++ b/src/novelai_python/sdk/ai/generate_image/__init__.py
@@ -274,6 +274,7 @@ def build(cls,
               mask: Union[str, bytes] = None,
               controlnet_model: Union[ControlNetModel, str] = None,
               controlnet_condition: str = None,
+              sm_dyn: bool = False,
               **kwargs
               ):
         """
@@ -298,6 +299,7 @@ def build(cls,
         :param mask: Inpainting mask
         :param controlnet_model: 控制网络模型
         :param controlnet_condition: 控制网络条件
+        :param sm_dyn: 是否使用动态sm
         :return: self
         """
         assert isinstance(prompt, str)
@@ -316,7 +318,8 @@ def build(cls,
             "add_original_image": add_original_image,
             "mask": mask,
             "controlnet_model": controlnet_model,
-            "controlnet_condition": controlnet_condition
+            "controlnet_condition": controlnet_condition,
+            "sm_dyn": sm_dyn,
         })
         if reference_mode:
             kwargs.update({