did you manage to install the Spacetimeformer so it works ? I keep having the problem with a torch version. Documentation says it requires Pytorch 1.11.0, but that version does not have torch.get_autocast_dtype() attribute and I can't find the answer to this problem. Any attempt to run:
python train.py model dset - h
crashes with this error.