Can't run with multiple gpu

Hello, thanks for the great work!

I refer to the `example_code/example_chat.py` to run the newest InternLM-XComposer-2.5 model using 4 NVIDIA 4090 GPUs. But still meet the OOM problem. It seems that although the weights are divided successfully, the first gpu always runs into OOM when `model.chat` is called.

Any response will be greatly appreciated!