Diffusion-based-Fluid-Super-resolution
Diffusion-based-Fluid-Super-resolution copied to clipboard
Fix hard-coded diffusion model type in trainer
I was getting an error when trying to train with the model type "simple", due to hard-coded ConditionalDiffusion. I added a check to see which model to use.