Expected A Cuda Device Type For Generator But Found Cpu