InternLM / InternLM/xtuner

如何实现多张卡共同存放单个模型

Open
#898 1 comment 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
5.2k
Forks
448
Avg merge
3d 15h
Merged PRs (30d)
26

Description

背景:
模型internlm2.5-7b-chat
单机4卡A10,单卡24GB

问题:
1. xtuner能实现模型切分么?即多个卡共用一个模型,而不是在每个卡上都单独加上一个模型再去微调训练,这样的话很容易会显存不足;能做到说模型切分到了某张卡运行,其余的卡加载数据训练吗?

2. 按官方要求的多轮对话格式训练时,其中的单个message json就达到了几百个,且暂时不拆分数据的情况下,在显存充足的情况下,xtuner能支持到多长(100K/500K)的上下文训练?多轮对话中的单个message json大小长度,有限制么?
![image](https://github.com/user-attachments/assets/9e2d63c4-3c3a-4412-81df-d0980eb4fc3b)

3. 如下图所示,整个微调数组中,只有一个messages 能训练吗?尽管messages内有多个键值对数据。
![image](https://github.com/user-attachments/assets/4af33461-d7bd-433e-bc05-286a7fd1f523)

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.