Multi GPU sampling when using Pre-trained VQ-VAE and upsamplers.

[ObscuraDK]

When I run multiple GPU sampling from a prior using the pre-trained VQ-VAE and upsamplers, the whole program freezes and Linux exit the program after a while. It works when I use a single GPU, and multi-GPU works when I use it on ny own small VQ-VAE and upsampler. But I get much better results using the pre-trained stuff.