flax-community
/

transformer-vae

Model card Files Files and versions

transformer-vae / model /config.py

Commit History

working saving & loading

7633929

Fraser commited on Jul 12, 2021

almost working training

c0f0a4a

Fraser commited on Jul 12, 2021

fix enc attention mask

11030c1

Fraser commited on Jul 11, 2021

add latent dim args

9416dd9

Fraser commited on Jul 10, 2021

passing most tests

74bc6c7

Fraser commited on Jul 10, 2021

will move to training args

d661afd

Fraser commited on Jul 7, 2021

closer to working

a516fcd

Fraser commited on Jul 7, 2021

getting closer with the code

e6109b4

Fraser commited on Jul 6, 2021

fixes

7bd06a8

Fraser commited on Jul 5, 2021

basic code

99674a9

Fraser commited on Jul 1, 2021