Phil Wang faebf4c8b8 from my vision transformer experience, dimension of attention head of 32 is sufficient for image feature maps

2022-04-20 11:40:32 -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

fix everything and make sure it runs end to end, document everything in readme for public

2022-04-13 18:05:25 -07:00

cli.py

give time tokens a surface area of 2 tokens as default, make it so researcher can customize which unet actually is conditioned on image embeddings and/or text encodings

2022-04-20 10:04:47 -07:00

dalle2_pytorch.py

from my vision transformer experience, dimension of attention head of 32 is sufficient for image feature maps

2022-04-20 11:40:32 -07:00

latent_diffusion.py

scaffold for latent diffusion

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

get ready for all the training related classes and functions

2022-04-12 09:54:50 -07:00