Merge pull request #30 from LlmKira/dev

fix(sdk): Add reference_image reference_strength etc.
LlmKira · Feb 24, 2024 · e8665b4 · e8665b4
2 parents b8e0202 + fa9a4f4
commit e8665b4
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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({