$ pytorch - diffusion - dream output = out. png target = cute dog riding a train