Phil Wang f19c99ecb0 fix decoder needing separate conditional dropping probabilities for image embeddings and text encodings, thanks to @xiankgx !

2022-04-30 08:48:05 -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

now completely OpenAI CLIP compatible for training

2022-04-29 13:05:01 -07:00

cli.py

some outlines to the eventual CLI endpoint

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

dalle2_pytorch.py

fix decoder needing separate conditional dropping probabilities for image embeddings and text encodings, thanks to @xiankgx !

2022-04-30 08:48:05 -07:00

optimizer.py

now completely OpenAI CLIP compatible for training

2022-04-29 13:05:01 -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

now completely OpenAI CLIP compatible for training

2022-04-29 13:05:01 -07:00

vqgan_vae.py

vqgan-vae codebook dims should be 256 or smaller

2022-04-28 08:59:03 -07:00