From dc7d7b88b271b520dce5849f2b5372b14d84667a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Junhao=20Zhuang=20=28=E5=BA=84=E4=BF=8A=E8=B1=AA=29?= <108931120+zhuang2002@users.noreply.github.com> Date: Fri, 19 Apr 2024 17:11:59 +0800 Subject: [PATCH] Update nodes.py --- nodes.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/nodes.py b/nodes.py index eb4df0f..c9d8686 100644 --- a/nodes.py +++ b/nodes.py @@ -479,6 +479,11 @@ class powerpaint_brushnet_sampler(brushnet_sampler): promptB_list = [] negative_promptB_list = [] + if task=='image-outpainting' or task == 'context-aware': + prompt = prompt + ' empty scene' + if task=='object-removal': + prompt = prompt + ' empty scene blur' + prompt_list = [prompt] * (B - len(prompt_list)) if len(prompt_list) < B else prompt_list n_prompt_list = [n_prompt] * (B - len(n_prompt_list)) if len(n_prompt_list) < B else n_prompt_list