primal_vae / modeling_primal_vae.py

Commit History

3 days ago

  1. Add 4x1 primal vae

    Verified signature
    b0b828d

7 days ago

  1. Add gradient checkpointing to the vae

    Verified signature
    5dcf26e

8 days ago

  1. patchify takes total compression, unpatchify infers it

    Verified signature
    8a432f7
  2. Primal VAE 2x1 and 2x2

    Verified signature
    bee1996