Mirror/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2024-06-07 21:20:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #14330 from AUTOMATIC1111/fix-extras-caption-BLIP

Browse Source 
fix extras caption BLIP
This commit is contained in:
AUTOMATIC1111 2023-12-16 16:41:40 +03:00 committed by GitHub
commit 3d9a0d9e4b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats 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)) captions.append(deepbooru.model.tag(pp.image))
if "BLIP" in option: 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]) pp.caption = ", ".join([x for x in captions if x])