Phil Wang 9878be760b have researcher explicitly state upfront whether to condition with text encodings in cascading ddpm decoder, have DALLE-2 class take care of passing in text if feature turned on

2022-04-26 09:47:09 -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

__init__.py

refactor blurring training augmentation to be taken care of by the decoder, with option to downsample to previous resolution before upsampling (cascading ddpm). this opens up the possibility of cascading latent ddpm

2022-04-22 11:09:17 -07:00

attention.py

start a file for all attention-related modules, use attention-based upsampling in the unets in dalle-2

2022-04-25 18:59:10 -07:00

cli.py

some outlines to the eventual CLI endpoint

2022-04-24 09:27:15 -07:00

dalle2_pytorch.py

have researcher explicitly state upfront whether to condition with text encodings in cascading ddpm decoder, have DALLE-2 class take care of passing in text if feature turned on

2022-04-26 09:47:09 -07:00

tokenizer.py

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

train.py

bring in the exponential moving average wrapper, to get ready for training

2022-04-25 19:24:13 -07:00

vqgan_vae.py

revert back to old upsampling, paper does not work

2022-04-26 07:39:04 -07:00