Skip to content

S
Stable Diffusion Webui
Commit 98c5fa92 authored by w-e-w's avatar w-e-w
Browse files
Options

fix extras caption BLIP 

#14328
parent 7428ce52
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
scripts/postprocessing_caption.py
View file @ 98c5fa92
...@@ -25,6 +25,6 @@ class ScriptPostprocessingCeption(scripts_postprocessing.ScriptPostprocessing): ...@@ -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])
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment