[FEATURE REQUEST] Support more caption methods in Dataset Maker

hollowstrawberry / kohya-colab

Accessible Google Colab notebooks for Stable Diffusion Lora training, based on the work of kohya-ss and Linaqruf

GNU General Public License v3.0

610 stars 92 forks source link

Open LOLSALT opened 1 month ago

LOLSALT commented 1 month ago

Hello, Could you support more caption methods(such as PaliGemma - Longprompt, JoyCaption and Florence2) in Dataset Maker? I think some of them looks like useful for some usage scenarios, if you could add them, thanks. Here are comparative of these caption methods difference on Flux: https://civitai.com/articles/6792/flux-style-captioning-differences-training-diary https://civitai.com/articles/7146/flux-style-captioning-differences-pt2-4-new-caption-tools-training-diary

hollowstrawberry commented 1 month ago

It would be good to implement these, I just can't put the time right now