RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu on transormer

Question

def nopeak_mask(size, opt):
    np_mask = np.triu(np.ones((1, size, size)),k=1).astype(\'uint8\')
    np_mask =Variable(torch.from_numpy(np_mask) == 0)
    if opt.devic