Support for Training with Custom Prompts

CompVis / zigma

A PyTorch implementation of the paper "ZigMa: A DiT-Style Mamba-based Diffusion Model" (ECCV 2024)

https://taohu.me/zigma

Apache License 2.0

281 stars 19 forks source link

Support for Training with Custom Prompts #23

Open rutuja1409 opened 2 months ago

rutuja1409 commented 2 months ago

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!