Merge pull request #15 from zhuang2002/main

Supplement powerpaint link & prompt
This commit is contained in:
Jukka Seppänen
2024-04-21 12:45:44 +03:00
committed by GitHub
2 changed files with 7 additions and 0 deletions
+2
View File
@@ -23,3 +23,5 @@ to `ComfyUI/models/brushnet`
Original repository:
https://github.com/TencentARC/BrushNet
https://github.com/zhuang2002/PowerPaint
+5
View File
@@ -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