Merge branch 'main' of https://huggingface.co/flax-community/transformer-vae into main

add wiki sentences

added placeholder checkpoint for the model

ran training

rearange files

working saving & loading

Merge branch 'main' of https://huggingface.co/flax-community/transformer-vae into main

fix reg loss

added the arg --model_name_or_path

added a script to test the loading of the model

added rng key to the eval_step

vmapped the mmd loss function

almost working training

small fixes

getting closer

fix enc attention mask

.

add latent dim args

Merge branch 'main' of https://huggingface.co/flax-community/transformer-vae into main

passing most tests

Update .gitattributes

ALMOST WORKING

small fixes

use latest

shift decoder tokens right

fix

will move to training args

update todos & add steps

Merge remote-tracking branch 'origin/theo' into main

fixed the inputs for the reg_loss (still not tried it)

added the mmd loss to the loss function (not tried the whole thing yet)

fix tabbing

closer to working

add links

add save & eval steps

getting closer with the code

fixes

add todos

show where to add reg loss

fix

basic code

initial commit