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

fix extras caption BLIP
2023-12-16 16:41:40 +03:00 · 2023-12-16 16:41:40 +03:00 · 3d9a0d9e4b
parent 7428ce52ab 98c5fa9201
commit 3d9a0d9e4b
1 changed files with 1 additions and 1 deletions
--- a/scripts/postprocessing_caption.py
+++ b/scripts/postprocessing_caption.py
@ -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])