Commit 98c5fa92 authored by w-e-w's avatar w-e-w
Browse files

fix extras caption BLIP 

#14328
parent 7428ce52

scripts/postprocessing_caption.py

+1 −1
Original line number Diff line number Diff line
@@ -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])

CRA Git | Maintained and supported by SUSTech CRA and CCSE