James Betker ac57cdc794 Add scheduling to quantizer, enable cudnn_benchmarking to be disabled		2021-09-24 17:01:36 -06:00
..
__init__.py
diffusion_dvae.py	Add scheduling to quantizer, enable cudnn_benchmarking to be disabled	2021-09-24 17:01:36 -06:00
fp16_util.py	Support diffusion unet	2021-06-06 13:57:22 -06:00
gaussian_diffusion.py	Get diffusion_dvae ready for prime time!	2021-09-16 22:43:10 -06:00
losses.py
nn.py	Add unet_diffusion_vocoder	2021-08-31 14:38:33 -06:00
resample.py
respace.py	Add support for a gaussian-diffusion-based wave tacotron	2021-07-26 16:27:31 -06:00
rrdb_diffusion.py	Misc fixes for diffusion nets	2021-06-21 10:38:07 -06:00
unet_diffusion_vocoder.py	Set kernel_size in diffusion_vocoder	2021-09-01 08:33:46 -06:00
unet_diffusion.py	More checkpointing fixes	2021-09-16 23:12:43 -06:00
unet_latent_guide.py	Add meta-anomaly detection, colorjitter augmentation	2021-06-29 13:41:55 -06:00