This was actually unused code in the VAE model, only does spatial tiling though, but seams look better