Phil Wang 8c2015fd39 add ability for DALL-E2 to return PIL images with return_pil_images = True on forward, for those who have no clue about deep learning

2022-04-27 19:57:27 -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

cli.py

some outlines to the eventual CLI endpoint

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

dalle2_pytorch.py

add ability for DALL-E2 to return PIL images with return_pil_images = True on forward, for those who have no clue about deep learning

2022-04-27 19:57:27 -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

fix issue with ema class

2022-04-27 16:40:02 -07:00

vqgan_vae.py

attention-based upsampling claims unsupported by local experiments, removing

2022-04-27 07:51:04 -07:00