Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

facehires use secondary prompt if present

Browse Source
This commit is contained in:
Vladimir Mandic
2024-03-10 14:09:50 -04:00
parent 8620ba5310
commit 3be2b4bcfe
2 changed files with 13 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/face-details.py
+9
View File
@@ -103,6 +103,8 @@ class FaceRestorerYolo(face_restoration.FaceRestoration):
shared.opts.data['mask_apply_overlay'] = True
p = processing_class.switch_class(p, processing.StableDiffusionProcessingImg2Img)
shared.log.error(orig_p)
for face in faces:
if face.mask is None:
continue
@@ -115,6 +117,13 @@ class FaceRestorerYolo(face_restoration.FaceRestoration):
p.inpainting_mask_invert = 0
p.inpainting_fill = 1 # no fill
p.denoising_strength = orig_p.get('denoising_strength', 0.3)
p.styles = []
p.prompt = orig_p.get('refiner_prompt', '')
if len(p.prompt) == 0:
p.prompt = orig_p.get('all_prompts', [''])[0]
p.negative_prompt = orig_p.get('refiner_negative', '')
if len(p.negative_prompt) == 0:
p.negative_prompt = orig_p.get('all_negative_prompts', [''])[0]
# TODO facehires expose as tunable
p.mask_blur = 10
p.inpaint_full_res_padding = 15