pe.api.image.improved_diffusion_lib.gaussian_diffusion module

This code contains minor edits from the original code at https://github.com/openai/improved-diffusion/blob/main/improved_diffusion/gaussian_diffusion.py and https://github.com/openai/improved-diffusion/blob/main/improved_diffusion/script_util.py to support sampling from the middle of the diffusion process with start_t and start_image arguments.

pe.api.image.improved_diffusion_lib.gaussian_diffusion.create_gaussian_diffusion(*, steps=1000, learn_sigma=False, sigma_small=False, noise_schedule='linear', use_kl=False, predict_xstart=False, rescale_timesteps=False, rescale_learned_sigmas=False, timestep_respacing='')