Susie Sargsyan

Thanks for the answer @patil-suraj On TPU v4 if you use runtime_version=tpu-vm-v4-pt-1.13 the xla comes preinstalled, so no need to install it separately. I have tested the flax version, but...

Same error when I run ``` python3 main.py --train --base configs/latent-diffusion/lsun_bedrooms-ldm-vq-4_ss.yaml ``` Though I also had to modify `configs/first_stage_models/vq-f4/model.yaml` to be `models/first_stage_models/vq-f4/config.yaml`

Closing since the server seems to be back.

Have the same issue for several days