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