Support for Training with Custom Prompts Instead of Class IDs

[rutuja1409]

Hi @devzhk ,

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

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

Thank you!