Code needed to run a T5-VAE with flax.
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool
Model code for running a T5-VAE with flax.