Is there a way to configure this to train on Mel Spectrograms?

[Aman-Khokhar18]

I am trying to train the model based on my mel spectrogram dataset but I cant seem to find a way to configure the model for that

[explainingai-code]

Hello @Aman-Khokhar18 , Can you let me know which specific part you are having trouble configuring.
Ideally just updating the config with the right resolution and channels in https://github.com/explainingai-code/StableDiffusion-PyTorch/blob/main/config/celebhq.yaml#L3-L4 should be enough, is there an issue that you are facing after you did that ?