only changes in forward, p_losses and sample functions of diffusion model.
3d flow used gaussian pyramids and scale space warp, 2d flow only uses flow warp
compared to cold start problem, both the new versions use swin ir that is instantiated in the diffusion model and used to upsample lowres as ures which gives context instead of hres
added a version for 2d flow
only changes in forward, p_losses and sample functions of diffusion model.
3d flow used gaussian pyramids and scale space warp, 2d flow only uses flow warp
compared to cold start problem, both the new versions use swin ir that is instantiated in the diffusion model and used to upsample lowres as ures which gives context instead of hres