support negative prompt

okotaku · Oct 2, 2023 · 250fc9d · 250fc9d
1 parent 15e8913
commit 250fc9d
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 0 deletions.
diff --git a/diffengine/models/editors/ip_adapter/ip_adapter_xl.py b/diffengine/models/editors/ip_adapter/ip_adapter_xl.py
@@ -128,6 +128,7 @@ def infer(self,
                 The image prompt or prompts to guide the image generation.
             negative_prompt (`Optional[str]`):
                 The prompt or prompts to guide the image generation.
+                Defaults to None.
             height (`int`, *optional*, defaults to
                 `self.unet.config.sample_size * self.vae_scale_factor`):
                 The height in pixels of the generated image.

diff --git a/diffengine/models/editors/stable_diffusion/stable_diffusion.py b/diffengine/models/editors/stable_diffusion/stable_diffusion.py
@@ -122,6 +122,7 @@ def infer(self,
                 The prompt or prompts to guide the image generation.
             negative_prompt (`Optional[str]`):
                 The prompt or prompts to guide the image generation.
+                Defaults to None.
             height (`int`, *optional*, defaults to
                 `self.unet.config.sample_size * self.vae_scale_factor`):
                 The height in pixels of the generated image.

diff --git a/diffengine/models/editors/stable_diffusion_controlnet/stable_diffusion_controlnet.py b/diffengine/models/editors/stable_diffusion_controlnet/stable_diffusion_controlnet.py
@@ -117,6 +117,7 @@ def infer(self,
                 The condition image for ControlNet.
             negative_prompt (`Optional[str]`):
                 The prompt or prompts to guide the image generation.
+                Defaults to None.
             height (`int`, *optional*, defaults to
                 `self.unet.config.sample_size * self.vae_scale_factor`):
                 The height in pixels of the generated image.

diff --git a/diffengine/models/editors/stable_diffusion_xl/stable_diffusion_xl.py b/diffengine/models/editors/stable_diffusion_xl/stable_diffusion_xl.py
@@ -169,6 +169,7 @@ def infer(self,
                 The prompt or prompts to guide the image generation.
             negative_prompt (`Optional[str]`):
                 The prompt or prompts to guide the image generation.
+                Defaults to None.
             height (`int`, *optional*, defaults to
                 `self.unet.config.sample_size * self.vae_scale_factor`):
                 The height in pixels of the generated image.

diff --git a/diffengine/models/editors/stable_diffusion_xl_controlnet/stable_diffusion_xl_controlnet.py b/diffengine/models/editors/stable_diffusion_xl_controlnet/stable_diffusion_xl_controlnet.py
@@ -118,6 +118,7 @@ def infer(self,
                 The condition image for ControlNet.
             negative_prompt (`Optional[str]`):
                 The prompt or prompts to guide the image generation.
+                Defaults to None.
             height (`int`, *optional*, defaults to
                 `self.unet.config.sample_size * self.vae_scale_factor`):
                 The height in pixels of the generated image.