Transformers How to use valhalla/cogview-vqvae-test with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import CogViewVQVAE
model = CogViewVQVAE.from_pretrained("valhalla/cogview-vqvae-test", device_map="auto") 
