Update nodes.py

2024-04-19 17:11:59 +08:00
parent 02de73bcc6
commit dc7d7b88b2
1 changed files with 5 additions and 0 deletions
@@ -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