Hugging Face's logo

Fraser
/
transformer-vae

Model card Files Community
Edit model card
YAML Metadata Warning: empty or missing yaml metadata in repo card (https://huggingface.co/docs/hub/model-cards#model-card-metadata)

Transformer-VAE (WIP)

A PyTorch Transformer-VAE model.

Uses an MMD loss to prevent posterior collapse.

Will setup in the next month or so.

ToDo

  • Copy in old repo code.
  • Make a bunch of sample training runs.
  • Make an interpolation widget?
Downloads last month
0
Unable to determine this model's library. Check the docs .