Black image using float16

[benjamin-bertram]

I tried to set up a colab, because my device does not support bfloat, but so doesn't colab. But with float16 i get a black image. Is there anything i can prevent this?

Edit: float16 is not supported right now. it works when the dtype = torch.bfloat16 in line 16 is changed to dtype = torch.float32