SVI with Multiple gpus

Hi. I am trying to run svi using multiple gpu devices but I am getting OOM errors seemingly because the data is all being put on a single gpu. How do i spread this across? Any tips/configuration suggestions would be super helpful