VIT-VQGAN Run code on TPU VMs Once you've created a TPU VM, run python tpu_run.py Basically this SSH's into your TPU VM, installs dependencies, and then runs your command. You should modify the code in tpu_run.py and configuration in pretrain/configs/*.yaml.