Bing-su/adetailer
1
0
Fork 0
mirror of https://github.com/Bing-su/adetailer.git synced 2026-03-14 09:47:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix ADetailer prompt in PNG Info showing the original prompt when ADetailer S/R script is applied (#489)

Browse Source
This commit is contained in:
whiteSkar
2024-02-26 03:08:53 -08:00
committed by GitHub
parent 8f01dfdaa9
commit afb9cd0b66
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
scripts/!adetailer.py
View File

@@ -648,6 +648,16 @@ class AfterDetailerScript(scripts.Script):
if self.is_ad_enabled(*args_):
arg_list = self.get_args(p, *args_)
self.check_skip_img2img(p, *args_)
# The extra_generation_params has the ADetailer prompt that will be saved in the png info
# In case S/R script has been applied, we need to return the replaced prompt and not the original one
# batch_index exists on p if the p is for the Adetailer p.
# If the p is for the image before Adetailer, batch_index does not seem to exist.
if hasattr(p, "_ad_xyz_prompt_sr") and hasattr(p, "batch_index"):
replaced_positive_prompt, replaced_negative_prompt = self.get_prompt(p, arg_list[0])
arg_list[0].ad_prompt = replaced_positive_prompt[0]
arg_list[0].ad_negative_prompt = replaced_negative_prompt[0]
extra_params = self.extra_params(arg_list)
p.extra_generation_params.update(extra_params)
else: