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

#3
by vivekwagh2003 - opened

response = remove_pii_info(
model,
tokenizer,
prompt,256,
temperature=0.7)
getting the above error over here ....pls help

Sign up or log in to comment