mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-12-06 10:31:17 -08:00
Code Issues Projects Releases Packages Wiki Activity

fix extras caption BLIP

Browse source 
#14328
This commit is contained in:
w-e-w 2023-12-16 22:14:39 +09:00
parent 7428ce52ab
commit 98c5fa9201
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
scripts/postprocessing_caption.py
View file

@ -25,6 +25,6 @@ class ScriptPostprocessingCeption(scripts_postprocessing.ScriptPostprocessing):
captions.append(deepbooru.model.tag(pp.image))
if "BLIP" in option:
captions.append(shared.interrogator.generate_caption(pp.image))
captions.append(shared.interrogator.interrogate(pp.image.convert("RGB")))
pp.caption = ", ".join([x for x in captions if x])