Skip to content

Commit

Permalink
Merge pull request #14330 from AUTOMATIC1111/fix-extras-caption-BLIP
Browse files Browse the repository at this point in the history
fix extras caption BLIP
  • Loading branch information
AUTOMATIC1111 authored Dec 16, 2023
2 parents 7428ce5 + 98c5fa9 commit 3d9a0d9
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion scripts/postprocessing_caption.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,6 @@ def process(self, pp: scripts_postprocessing.PostprocessedImage, enable, option)
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])

1 comment on commit 3d9a0d9

@jaffster595
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Can confirm that this fixes the BLIP captioning error.

Please sign in to comment.