Fix hard-coded diffusion model type in trainer#16

jvwilliams23 wants to merge 1 commit intoBaratiLab:main_v1BaratiLab/Diffusion-based-Fluid-Super-resolution:main_v1from

jvwilliams23:main_v1jvwilliams23/Diffusion-based-Fluid-Super-resolution:main_v1Copy head branch name to clipboard