angellmethod

I can repo this but I think the issue in pytorch core from this issue I found: https://github.com/pytorch/pytorch/issues/84523

The simple workaround as far as this notebook goes looks like moving the detach().cpu() operation into `make_grid`. ``` grid = torchvision.utils.make_grid(x.detach().cpu()) grid_im = grid.permute(1, 2, 0).clip(0, 1) * 255 ```