zsyOAOA / ResShift

ResShift: Efficient Diffusion Model for Image Super-resolution by Residual Shifting (NeurIPS@2023 Spotlight, TPAMI@2024)
Other
870 stars 47 forks source link

Hi, I would like to ask the training step for other dataset #36

Open maxxxxxxcai opened 8 months ago

maxxxxxxcai commented 8 months ago

Thanks for the brilliant work!! For the other datasets, is that true to train the autoencoder first and then train the diffusion model?

zsyOAOA commented 8 months ago

There is no need to train the autoencoder. You can borrow the autoencoder from any latent diffusion model, such as stable diffsuion.