Support for Training with Custom Prompts

Hi @stefan-baumann @GrebenkovaO @dongzhuoyao @dimakot55 @m990130 ,

I would like to train a model using my custom dataset. Is there a way to provide a custom prompt for each image in webdataset format and for inference?

If this feature is not currently available, is there a plan to include it in any upcoming releases?

Thank you!