Support for Training with Custom Prompts Instead of Image IDs

[rutuja1409]

Hi @gasvn,

I would like to train a model using my custom dataset. However, I noticed that the current training process only supports using image IDs. Is there a way to provide a custom prompt for each image instead of using just the image ID?

If this feature is not currently available, is there a plan to include it in any upcoming releases?

Thank you!