Currently show_image_grid expects torch tensors.
If a numpy array should be displayed the array needs to be cast to torch tensors before:
images = torch.from_numpy(images)
Maybe it would be more convenient to have this check and cast inside the show_image_grid method