Ray Bell 44b319cb57 add missing import (#56)

2022-05-04 07:42:20 -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

simplify Decoder training for the public

2022-04-30 11:45:18 -07:00

cli.py

add missing import (#56)

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

dalle2_pytorch.py

add ability to turn on normformer settings, given @borisdayma reported good results and some personal anecdata

2022-05-02 11:33:15 -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_vqgan_vae.py

port over vqgan vae trainer

2022-05-01 08:09:15 -07:00

train.py

fix issue with mixed precision and gradient clipping

2022-05-02 09:20:19 -07:00

vqgan_vae.py

fix calculation of adaptive weight for vit-vqgan, thanks to @CiaoHe

2022-05-02 07:58:14 -07:00