Hi, I want to utilize this work for other data, but it seems that only Text2image training script is provided. How to train VQVAE models?