Phil Wang f4016f6302 allow for overriding use of EMA during sampling in decoder trainer with use_non_ema keyword, also fix some issues with automatic normalization of images and low res conditioning image if latent diffusion is in play

2022-05-16 11:18:30 -07:00

..

__init__.py

Migrate to text-conditioned prior training (#95)

2022-05-15 20:16:38 -07:00

decoder_loader.py

Add a dataloader for training the decoder (#57)

2022-05-05 07:08:45 -07:00

embedding_wrapper.py

Migrate to text-conditioned prior training (#95)

2022-05-15 20:16:38 -07:00

simple_image_only_dataloader.py

allow for overriding use of EMA during sampling in decoder trainer with use_non_ema keyword, also fix some issues with automatic normalization of images and low res conditioning image if latent diffusion is in play

2022-05-16 11:18:30 -07:00