Hugging Face's logo

flax-community
/
transformer-vae

Model card Files
xet
 Community
transformer-vae / model

Commit History

  1. working saving & loading

    Fraser
    7633929

  2. almost working training

    Fraser
    c0f0a4a

  1. small fixes

    Fraser
    9a1e44d

  2. fix enc attention mask

    Fraser
    11030c1

  1. .

    Fraser
    c377535

  2. add latent dim args

    Fraser
    9416dd9

  3. passing most tests

    Fraser
    74bc6c7

  1. ALMOST WORKING

    Fraser
    92871f7

  2. small fixes

    Fraser
    6f4a0d9

  1. shift decoder tokens right

    Fraser
    1df308b

  2. fix

    Fraser
    9c2b63e

  3. will move to training args

    Fraser
    d661afd

  4. fix tabbing

    Fraser
    568b60a

  5. closer to working

    Fraser
    a516fcd

  1. getting closer with the code

    Fraser
    e6109b4

  1. fixes

    Fraser
    7bd06a8

  1. add todos

    Fraser
    db1be99

  2. basic code

    Fraser
    99674a9