How to use flax-community/t5-vae-wiki with Transformers:
# Load model directly from transformers import T5VaeForAutoencoding model = T5VaeForAutoencoding.from_pretrained("flax-community/t5-vae-wiki", dtype="auto")