Is it possible to train these models with custom data?

@Victarry I would like to generate samples using my own custom dataset. More precisely, I have some labels that I would like to feed into the network along with my images. Please can you advise which model I can use and how I must adapt the code?