DMAD 项目团队开源了用于联合音视频生成的 MiniMax-H3 4 步学生模型,同时发布面向图像、视频和音视频的少步蒸馏方法、训练与推理代码。

在 MiniMax-H3-33B 上,DMAD 将 50 步教师模型蒸馏为 4 步学生模型,减少生成过程中的去噪迭代次数,而不是将模型参数压缩为小模型。

该方法通过 critic(判别器)直接学习指导学生更新的分布差异,省去持续拟合学生分布的辅助扩散模型,并在 ImageNet、SDXL、Wan2.1 与 MiniMax-H3 上验证。

开源版本支持 1344×768 视频与 32 kHz 原生立体声音频,并提供 8 GB 显存消费级显卡可用的约 5 秒 ComfyUI 工作流。

 

开源地址:

效果展示

折纸龙飞越纸山

纸龙展开棱角分明的双翼,飞越一座座折纸山峰。该样例由 MiniMax-H3-33B 的 DMAD 4 步学生模型生成,属于联合音视频生成。

“Paper comes to life”样例的四个关键帧:

 

 

松鼠走出照片

手把松果放在照片旁,松鼠从纸上的树洞探出身,伸爪触碰松果。该样例由 MiniMax-H3-33B 的 DMAD 4 步学生模型生成,以照片与立体角色的互动组织动作序列。

“A story comes to life”样例的四个关键帧:

 

 

壁画前的街头交谈

壁画前,两名男子交谈,年轻人转头露出笑容。MiniMax-H3-33B 4 步音视频样例的官方提示词同时指定人物对白、城市环境声与背景音乐。

“Street conversation”样例的四个关键帧:

 

 

霓虹街道上的双犬

萨摩耶与金毛穿过霓虹街道,蓝紫色灯光映在毛发与路面上。该样例使用 Wan2.1-T2V-14B 的 DMAD 4 步学生模型,属于文生视频,不含生成音轨。

官网“After dark”样例的四个关键帧:

 

 

模型框架

DMAD(Distribution Matching as Adversarial Distillation,基于对抗蒸馏的分布匹配)解决的是扩散模型需要多次串行去噪的问题。它不要求学生逐条复现教师的采样轨迹,而是让少步学生生成的样本分布接近目标分布;

MiniMax-H3 版本沿用 33B 基座并通过 LoRA 训练,主要减少采样步数,而非压缩模型参数规模。

 

从分别估计两个分数到直接学习分布差异

DMD 的学生更新依赖目标分布与学生分布的 score(对数密度梯度)之差:教师提供目标 score,另一个辅助扩散模型持续追踪学生分布。DMAD 改为训练二分类判别器,区分目标样本和学生样本。

在理想最优判别器条件下,其 logit 等于两种分布的对数密度比;固定该判别器,对输入求梯度便能得到同一个 score 差,形成局部生成器更新方向上的等价。直观上,前者先估计两条路线再求差,后者直接学习该往哪儿修正,因此不再需要辅助 score 拟合。

共享骨干、双头监督

两个判别头共享特征骨干:教师头区分教师样本与学生样本,真实数据头区分真实样本与学生样本。判别器用二元交叉熵训练,学生则用两个 logit 的加权线性损失更新;更新学生时冻结判别器参数,但保留经过其输入的反向梯度。教师头和真实数据头承担不同监督目标,不是分别处理音频、视频的两个分支。

按噪声区间动态分配教师监督

教师样本与真实数据的差异并不在所有噪声水平上相同。DMAD 使用真实数据头在两类样本上的平均 logit 差作为经验信号,以衰减系数 0.99 的指数移动平均追踪各噪声区间;差距较小的区间获得更高教师权重。权重归一化后参与学生损失,但统计量和权重不参与反向传播。

离线教师样本,减少在线蒸馏负担

教师仅提供预生成并缓存的样本,训练时无需在线调用教师,也不再拟合学生 score。论文采用每次学生更新对应 1 次判别器更新,DMD2 对比配置为 5 次。在 4 张 H200、MiniMax-H3-33B rank-128 LoRA 配置下,每次学生更新及关联判别器更新耗时从 818.1 秒降至 233.3 秒,约为 3.51 倍训练更新加速。

 

模型使用

环境与显存配置

命令行推理按官方仓库使用 Python 3.12、CUDA 12.6 与固定版本的 PyTorch 和 Diffusers,需准备支持 CUDA 的 NVIDIA 显卡。低显存方案通过权重卸载与分块计算控制显存占用,并会使用较多系统内存;

ComfyUI 低显存工作流明确推荐 64 GB 系统内存。命令行所需基础模型文件约 170 GB,需预留足够磁盘空间;使用前请按基础模型页要求阅读并接受相应许可。

使用方式显存配置视频长度
ComfyUI 工作流8GB约5秒
ComfyUI 工作流16GB及以上约15秒
命令行低显存推理8GB约5秒
命令行低显存推理12GB及以上约15秒

以上配置均对应 1344×768 有声视频。低显存配置会使用较多系统内存,实际生成耗时取决于显卡、存储和具体设置;4 步描述的是去噪步数,不代表 4 秒完成生成。

安装环境

以下命令在 DMAD 仓库根目录运行。requirements.txt 已固定相关依赖;modelscope-hub 提供下载模型所用的 modelscope 命令。

git clone https://github.com/Yzmblog/DMAD.git
cd DMAD
 
conda create -n dmad_h3 python=3.12 -y
conda activate dmad_h3
 
python -m pip install \
 torch==2.11.0 torchvision==0.26.0 torchaudio==2.11.0 \
 --index-url https://download.pytorch.org/whl/cu126
 
python -m pip install -r requirements.txt
python -m pip install -U modelscope-hub

 

从魔搭下载模型

先下载 MiniMax-H3 命令行推理所需的基础模型,再下载 DMAD 的 4 步 LoRA。下面排除了本次命令行示例不使用的 FL2VA、Ref2VA 和 transformer_ref 分区。

modelscope download MiniMax/MiniMax-H3 \
  --local-dir ./models/MiniMax-H3 \
  --exclude "FL2VA/*" "Ref2VA/*" "transformer_ref/*"
 
modelscope download ZhengmingYu-Community/DMAD \
  --local-dir ./ckpt \
  --include "minimax_h3/dmad_minimax_h3_4step_lora_critic.safetensors"

 

8 GB 显存生成约 5 秒有声视频

先使用仓库提供的提示词跑通样例。下面采用 4 步、固定种子和 124 帧,在 24 fps 下约为 5.2 秒,并启用 8 GB 低显存配置。

python inference.py \
  --model-dir ./models/MiniMax-H3 \
  --lora ./ckpt/minimax_h3/dmad_minimax_h3_4step_lora_critic.safetensors \
  --prompt-file prompts/dmad_sweater.txt \
  --steps 4 --seed 42 \
  --height 768 --width 1344 --num-frames 124 \
  --low-vram 8 \
  --output-dir ./outputs/dmad_5s

输出目录包含有声视频 0000.mp4、提示词 0000.txt 和运行参数 settings.json。跑通后,可以在命令中换成自己的提示词文件。

12 GB 显存生成约 15 秒视频

使用至少 12 GB 显存时,将帧数设为 362,并启用 12 GB 低显存配置。其余设置保持 4 步与 1344×768,输出约 15.1 秒有声视频。

python inference.py \
  --model-dir ./models/MiniMax-H3 \
  --lora ./ckpt/minimax_h3/dmad_minimax_h3_4step_lora_critic.safetensors \
  --prompt-file prompts/dmad_sweater.txt \
  --steps 4 --seed 42 \
  --height 768 --width 1344 --num-frames 362 \
  --low-vram 12 \
  --output-dir ./outputs/dmad_15s

 

ComfyUI 工作流

如果习惯用 ComfyUI,可以使用现有 ComfyUI 环境,导入仓库中的 comfyui/workflows/dmad_h3_4step_5s_8gb_guitar.json。项目也提供嵌入了工作流的示例 MP4,可直接拖入 ComfyUI。按工作流中的链接下载基础模型,并通过 ComfyUI Manager 安装 H3-Optimizations 节点包。

ComfyUI 使用专用格式的 LoRA。例如,下面从魔搭下载 full_critic 的 ComfyUI 版本,下载后将文件放入 ComfyUI/models/loras/。

modelscope download ZhengmingYu-Community/DMAD \
  --local-dir ./ckpt \
  --include "minimax_h3/dmad_minimax_h3_4step_full_critic_comfyui.safetensors"

 

DMAD 的 8 GB ComfyUI 工作流,4 步生成约 5 秒、1344×768 有声视频:

 

ComfyUI 工作流与配置说明
https://github.com/Yzmblog/DMAD/tree/main/comfyui

Logo

ModelScope旨在打造下一代开源的模型即服务共享平台,为泛AI开发者提供灵活、易用、低成本的一站式模型服务产品,让模型应用更简单!

更多推荐