SETUP DATA & MODEL TRAINING INFERENCE config params batch add init timesteps weights ckpt images README / Setup Dataset Pipeline Model Definition Noise Schedule Training Loop Sampler / Denoiser Sample Output