OpenGVLab

/

InternVL-Chat-V1-5

like 402

Follow

OpenGVLab 313

I'm seeing errors like this when using device_map=auto

File "modules/transformers_modules/OpenGVLab/InternVL-Chat-V1-5-Int8/0d0a0d47a718aaef231cb00c6cf9ce3fb905269b/modeling_internvl_chat.py", line 348, in generate
   input_embeds[selected] = vit_embeds.reshape(-1, C)
   ~~~~~~~~~~~~^^^^^^^^^^
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cuda:1!
  Gracefully stopping... (press Ctrl+C again to force)

Is this known? Is there a fix?