cog-sdxl
cog-sdxl copied to clipboard
Fix caption token casing not working with punctuation
When using token prefixes like A photo of TOK,
, the BLIP caption outputs no longer received the substitution token uppercasing treatment because of the punctuation. Using re's word boundary search, which will support various edge cases like beginning and ending of words, along with punctuation like commas.
/cc @cloneofsimo