Phil Wang eaf9d9fdc4 unet needs to be conditioned on log(snr) in p_mean_variance for continuous time gaussian diffusion

2022-06-08 00:41:41 -07:00

..

__init__.py

get working version of gaussian diffusion with continuous time (only beta linear schedule for now, but will eventually contain alpha cosine schedule as well as parameterized, learned monotonic MLP)

2022-06-07 15:59:29 -07:00

continuous_time_gaussian_diffusion.py

unet needs to be conditioned on log(snr) in p_mean_variance for continuous time gaussian diffusion

2022-06-08 00:41:41 -07:00

denoising_diffusion_pytorch.py

use a non-sinusoidal embedded condition for continuous time gaussian diffusion conditioned on log(snr)

2022-06-07 21:15:27 -07:00

learned_gaussian_diffusion.py

final cleanup

2022-05-12 13:58:16 -07:00

weighted_objective_gaussian_diffusion.py

add gaussian diffusion where model predicts both noise and x_start, with a learned weighting between the two (experimental)

2022-05-13 13:56:32 -07:00