how to have batch size larger than 1 and multi gpu when sampling

[TMullerSG]

Hi, when i changed batch size larger than 1, i get error in gaussian_diffusion.py, assert t.shape == (B,). and you mentioned can use num_samples to control number of test images, but in the code , num_samples only appear in
while len(all_images) * args.batch_size < args.num_samples: and all_images is nothing but a empty list, how does num_samples change anything?