Slackbot
06/26/2023, 4:55 PMlarme (shenyang)
06/27/2023, 4:26 PMBharath Krishna
06/29/2023, 11:40 PMRuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when
checking argument for argument index in method wrapper_CUDA__index_select)Bharath Krishna
06/29/2023, 11:41 PM{'input_ids': tensor([[ 259, 74606, 207850, 74295]], device='cuda:0'), 'attention_mask': tensor([[1, 1, 1, 1]], device='cuda:0')}
which is using cuda but model_runner is not using cuda.larme (shenyang)
07/03/2023, 7:52 PMbentoml.transformers it should automatically lift tensor to cuda. Your error message means some tensor are lifted to cuda but some are not. Very strange behavior.