James Betker 46e9f62be0 Add unet with latent guide ... This is a diffusion network that uses both a LQ image and a reference sample HQ image that is compressed into a latent vector to perform upsampling The hope is that we can steer the upsampling network with sample images.		2021-06-26 11:02:58 -06:00
..
__init__.py
fp16_util.py	Support diffusion unet	2021-06-06 13:57:22 -06:00
gaussian_diffusion.py	A few fixes:	2021-06-05 13:40:32 -06:00
losses.py
nn.py
resample.py
respace.py
rrdb_diffusion.py	Misc fixes for diffusion nets	2021-06-21 10:38:07 -06:00
unet_diffusion.py	Pass a corruption factor through the dataset into the upsampling network	2021-06-07 09:13:54 -06:00
unet_latent_guide.py	Add unet with latent guide	2021-06-26 11:02:58 -06:00