make_grid expects a 4D tensor or a list of of images as described in the docs 1:

tensor (Tensor or list) – 4D mini-batch Tensor of shape (B x C x H x W) or a list of images all of the same size.

while you are trying to pass a numpy array to it.