lucidrains dab2f74650 fix self_attn type on unetconfig

2023-10-18 21:02:50 -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

dataloaders

Prior updates (#211)

2022-07-20 18:04:26 -07:00

__init__.py

new einops is torch compile friendly

2023-10-18 15:45:09 -07:00

cli.py

add missing import (#56)

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

dalle2_pytorch.py

remove einops exts for better pytorch 2.0 compile compatibility

2023-04-20 07:05:29 -07:00

optimizer.py

lower beta2 for adam down to 0.99, based on https://openreview.net/forum?id=2LdBqxc1Yv

2022-06-03 10:26:28 -07:00

tokenizer.py

make youtokentome optional package, due to reported installation difficulties

2022-06-01 09:25:35 -07:00

trackers.py

Re-introduced change that was accidentally rolled back (#212)

2022-07-20 23:01:19 -07:00

train_configs.py

fix self_attn type on unetconfig

2023-10-18 21:02:50 -07:00

trainer.py

bug fix cosine annealing optimizer in prior trainer (#262)

2022-11-11 12:15:13 -08:00

utils.py

Overhauled the tracker system (#172)

2022-07-01 09:39:40 -07:00

version.py

fix self_attn type on unetconfig

2023-10-18 21:02:50 -07:00

vqgan_vae_trainer.py

EMA for vqgan vae comes from ema_pytorch now

2022-06-20 15:29:08 -07:00

vqgan_vae.py

remove einops exts for better pytorch 2.0 compile compatibility

2023-04-20 07:05:29 -07:00