Dreambooth training with image captions

[alireza18878]

Hi all,

I would like to fine-tune a Dreambooth model with my own dataset, and I want the model to learn the captions (description of pictures) as well.
I tried using Datasets but it doesn't recognize the metadata.csv file.
Can you please let me know if it is possible to train a Dreambooth model while incorporating the picture descriptions (captions)?

Thanks

[2016312357]

You can modify Dreambooth-Stable-Diffusion-main\ldm\data\personalized.py to read your own data.