How to use custon dataset in train_dreambooth_flux.py.

[rooooc]

Hi. what if i want to train two images with two different prompts. somethink like m1.jpeg , m1.txt ; m2.jpeg, m2.txt.
the default example only shows all images share one instant prompt. thanks for the help!

[hlky]

Look at options --dataset_name --image_column --caption_column, you may need to prepare the dataset so that datasets understands it.

cc @linoytsaban do you have any examples for this?

[linoytsaban]

Yes exactly you can use --dataset_name --image_column --caption_column to pass images with correspond captions,
this example should make it clear :)
let us know if you have further questions

[hlky]

Thanks @linoytsaban

@rooooc I've added documentation and a script to create metadata.jsonl from your image text pairs here #10664