Ming-Lite-Omni v1.5 使用指南

目录

简介

Ming-Lite-Omni v1.5 是对 Ming-lite-omni 的全模态能力的全面升级。它在图像文本理解、文档理解、视频理解、语音理解和合成以及图像生成和编辑等任务中的性能显著提升。基于 Ling-lite-1.5,Ming-lite-omni v1.5 总共有 203 亿个参数,其 MoE(专家混合)部分有 30 亿个活跃参数。与行业领先的模型相比,它在各种模态基准测试中展示了高度竞争的结果。

参考实现

url=https://github.com/inclusionAI/Ming
commit_id=d97e2f31467298674426539915a146d88a814925

变更记录

2025.07.30: 支持 Ming-Lite-Omni v1.5 图文理解任务训练。

环境安装

【模型开发时推荐使用配套的环境版本】

请参考安装指南

注意:Python版本推荐3.10,torch和torch_npu版本推荐2.7.1版本

1. 仓库拉取

git clone --branch 2.3.0 https://gitcode.com/Ascend/MindSpeed-MM.git
git clone https://github.com/inclusionAI/Ming.git

cd Ming
git checkout d97e2f3
mkdir -p logs data ckpt
cp -r ../MindSpeed-MM/examples/ming/* ./
cd ..

2. 环境搭建

# python3.10
conda create -n test python=3.10
conda activate test

# 安装torch和torch_npu
pip install torch-2.7.1-cp310-cp310-*.whl
pip install torch_npu-2.7.1*.manylinux2014_aarch64.whl

# 安装MindSPeed MM依赖
pip install -r MindSpeed-MM/examples/ming/requirements.txt

权重下载

从Hugging Face等网站下载开源模型权重

数据集准备及处理

以coco2017数据集为例,可以参考此处 下载和预处理数据集,预处理完后,数据格式如下:

[
 {
     "messages": [
     {
         "content": "<image>Who are they?",
         "role": "user"
     },
     {
         "content": "They're Kane and Gretzka from Bayern Munich.",
         "role": "assistant"
     }
     ],
     "images": [
     "mllm_demo_data/1.jpg"
     ]
 },
 ... ...
]

训练

1. 准备工作

配置脚本前需要完成前置准备工作,包括:环境安装权重下载数据集准备及处理,详情可查看对应章节。

2. 启动训练

  1. 以图文理解的微调任务为例,可根据实际情况修改启动脚本的配置,以下配置必须修改:
DATA_PATH="./data/mllm_format_llava_instruct_data.json" # 数据集的文件
DATA_DIR="./data" # 数据集依赖图文等文件的目录
PROCESSOR_PATH="." # Ming代码仓路径,默认是当前目录(即 Ming 目录)
LOAD_PATH="./ckpt/Ming-Lite-Omni-1.5" # Hugging Face下载的权重路径
  1. 根据使用机器的情况,修改NNODESNPUS_PER_NODE配置, 例如单机 A2 可设置NNODES为 1 、NPUS_PER_NODE为8;

  2. 为保证代码安全,配置trust_remote_code默认为False,用户需要在启动脚本中使能--trust-remote-code,并且确保自己下载的模型和数据的安全性。

  3. 上述注意点修改完毕后,可启动脚本开启训练:

cd Ming
bash finetune_vl.sh

环境变量声明

环境变量 描述 取值说明
ASCEND_SLOG_PRINT_TO_STDOUT 是否开启日志打印 0: 关闭日志打屏
1: 开启日志打屏
ASCEND_GLOBAL_LOG_LEVEL 设置应用类日志的日志级别及各模块日志级别,仅支持调试日志 0: 对应DEBUG级别
1: 对应INFO级别
2: 对应WARNING级别
3: 对应ERROR级别
4: 对应NULL级别,不输出日志
TASK_QUEUE_ENABLE 用于控制开启task_queue算子下发队列优化的等级 0: 关闭
1: 开启Level 1优化
2: 开启Level 2优化
COMBINED_ENABLE 设置combined标志。设置为0表示关闭此功能;设置为1表示开启,用于优化非连续两个算子组合类场景 0: 关闭
1: 开启
CPU_AFFINITY_CONF 控制CPU端算子任务的处理器亲和性,即设定任务绑核 设置0或未设置: 表示不启用绑核功能
1: 表示开启粗粒度绑核
2: 表示开启细粒度绑核
HCCL_CONNECT_TIMEOUT 用于限制不同设备之间socket建链过程的超时等待时间 需要配置为整数,取值范围[120,7200],默认值为120,单位s
PYTORCH_NPU_ALLOC_CONF 控制缓存分配器行为 expandable_segments:<value>: 使能内存池扩展段功能,即虚拟内存特征
HCCL_EXEC_TIMEOUT 控制设备间执行时同步等待的时间,在该配置时间内各设备进程等待其他设备执行通信同步 需要配置为整数,取值范围[68,17340],默认值为1800,单位s
ACLNN_CACHE_LIMIT 配置单算子执行API在Host侧缓存的算子信息条目个数 需要配置为整数,取值范围[1, 10,000,000],默认值为10000
TOKENIZERS_PARALLELISM 用于控制Hugging Face的transformers库中的分词器(tokenizer)在多线程环境下的行为 False: 禁用并行分词
True: 开启并行分词
MULTI_STREAM_MEMORY_REUSE 配置多流内存复用是否开启 0: 关闭多流内存复用
1: 开启多流内存复用
NPU_ASD_ENABLE 控制是否开启Ascend Extension for PyTorch的特征值检测功能 设置0或未设置: 关闭特征值检测
1: 表示开启特征值检测,只打印异常日志,不告警
2:开启特征值检测,并告警
3:开启特征值检测,并告警,同时会在device侧info级别日志中记录过程数据
ASCEND_LAUNCH_BLOCKING 控制算子执行时是否启动同步模式 0: 采用异步方式执行
1: 强制算子采用同步模式运行
NPUS_PER_NODE 配置一个计算节点上使用的NPU数量 整数值(如 1, 8 等)