Output shape should be same as the input shape in conditional image generation

[mamyou96]

https://github.com/Janspiry/Image-Super-Resolution-via-Iterative-Refinement/blob/01d27a7cbfa8502be1d8dbd4ee02fcbd5e44389d/model/ddpm_modules/diffusion.py#L218

In this line it requires the input shape be same as the output shape in conditional image generation that is not true. Instead of that, the output shape from the config file can be used to create the initial random image.

[LT1st]

You are right, I should have seen this before I spend days on it. Thanks