Phil Wang 708809ed6c lower beta2 for adam down to 0.99, based on https://openreview.net/forum?id=2LdBqxc1Yv

2022-06-03 10:26:28 -07:00

..

data

bring in the simple tokenizer released by openai, but also plan on leaving room for custom tokenizer with yttm

2022-04-12 09:23:17 -07:00

dataloaders

Improve Readability (#133)

2022-06-01 13:28:02 -07:00

__init__.py

fix version

2022-05-30 11:06:34 -07:00

cli.py

add missing import (#56)

2022-05-04 07:42:20 -07:00

dalle2_pytorch.py

allow for an option to constrain the variance interpolation fraction coming out from the unet for learned variance, if it is turned on

2022-06-03 09:34:57 -07:00

optimizer.py

lower beta2 for adam down to 0.99, based on https://openreview.net/forum?id=2LdBqxc1Yv

2022-06-03 10:26:28 -07:00

tokenizer.py

make youtokentome optional package, due to reported installation difficulties

2022-06-01 09:25:35 -07:00

trackers.py

make youtokentome optional package, due to reported installation difficulties

2022-06-01 09:25:35 -07:00

train_configs.py

make sure diffusion prior can be instantiated from pydantic class without clip

2022-05-26 08:47:30 -07:00

trainer.py

fix update_every within EMA

2022-06-03 10:21:05 -07:00

utils.py

make youtokentome optional package, due to reported installation difficulties

2022-06-01 09:25:35 -07:00

version.py

lower beta2 for adam down to 0.99, based on https://openreview.net/forum?id=2LdBqxc1Yv

2022-06-03 10:26:28 -07:00

vqgan_vae_trainer.py

unet_number on decoder trainer only needs to be passed in if there is greater than 1 unet, so that unconditional training of a single ddpm is seamless (experiment in progress locally)

2022-05-16 09:17:17 -07:00

vqgan_vae.py

remove convnext blocks, they are illsuited for generative work, validated by early experimental results at https://github.com/lucidrains/video-diffusion-pytorch

2022-05-05 07:07:21 -07:00