- 人工智能
- 大模型
- 多模态
- 媒体生成
- 模型量化
【免费下载链接】SenseNova-U1.5-8B-MoT-8step-4bit
SenseNova-U1.5-8B-MoT-8step-4bit 是商汤(SenseTime)统一多模态旗舰模型 SenseNova-U1.5-8B-MoT 的Apple Silicon(MLX)运行时工件,一条命令即可在 Mac 上完成文生图、指令编辑与视觉问答(VQA)。它内置官方 8 步蒸馏 LoRA 与 4bit 流式量化,配合 MLX-Swift 运行时实现零转换加载:权重文件布局与运行时内存布局完全一致,加载峰值内存≈常驻内存,16GB 级 Mac 也能跑得动 ⚡
模型是什么:8 步蒸馏 + 4bit 量化的 8B 多模态旗舰
上游模型采用NEO-unify Mixture-of-Transformers(MoT)架构,在像素空间用 rectified flow 直接生成图像,不依赖 VAE,同时覆盖三大任务:
- 🎨文生图(T2I):纯文本描述生成图像
- ✂️指令编辑:按自然语言修改已有图像
- 🔍视觉问答(VQA):对图像内容提问并回答
本工件在原始权重之上做了两件事(见 README.md):
- 合并官方 8 步蒸馏 LoRA(
lora_merged: SenseNova-U1.5-8B-MoT-LoRA-8step,config.json)——把原本需要 50 步的去噪压缩到8 步,速度大幅提升; - 4bit 流式量化(group size 64,config.json)——把内存占用压到 16GB 机型可承载的范围。
核心规格一览:
| 配置项 | 数值 | 说明 |
|---|---|---|
| LLM 主干 | Qwen3,42 层,hidden 4096 | 理解与生成共享骨干 |
| 注意力 | 32 头 / 8 KV 头(GQA),head_dim 128 | 显存友好的分组注意力 |
| 词表 | 151936,最大上下文 262144 | 与上游分词逐字节一致 |
| 视觉编码器 | NEOVisionModel,hidden 1024 | patch_size 16,直接接像素头 |
| 量化 | 4bit,group_size 64 | MLX 约定quantization块 |
仓库文件清单:约 11GB 权重的分片结构
整个仓库结构非常干净,权重分 3 个 safetensors 分片存放,总量11,768,447,360 字节(约 11GB),记录在 model.safetensors.index.json 的metadata.total_size中:
| 文件 | 作用 |
|---|---|
| model-00001-of-00003.safetensors | 权重分片 1(含 fm 头、第 0/10~19 层等) |
| model-00002-of-00003.safetensors | 权重分片 2(第 2~9 层、20~41 层主体) |
| model-00003-of-00003.safetensors | 权重分片 3(第 5 层生成流、42 层后归一化、视觉编码器) |
| model.safetensors.index.json | 每个权重张量 → 所在分片的映射表 |
| config.json | 架构、量化、蒸馏等全部元信息 |
| tokenizer.json / vocab.json / merges.txt | 分词器(由上游 vocab/merges 生成,tokenization 逐字节一致) |
| added_tokens.json / special_tokens_map.json | 特殊 token 定义 |
| tokenizer_config.json | 分词器配置(Qwen2Tokenizer,含 thinking 对话模板) |
零转换加载原理:磁盘布局 = 内存布局
这是本工件最核心的设计,也是标题里"零转换"的由来。
常规流程是"三段式"的:先从磁盘读入 bf16 原始权重 → 在 CPU/GPU 里转换成目标框架的内存布局 → 量化/重排成推理格式。每一步都会产生临时缓冲区,导致加载峰值内存远高于最终常驻内存,内存紧张时甚至直接 OOM。
本工件的做法(见 README.md 的 Runtime 一节):
- 权重直接以运行时
sensenova-u1-swift(MLX-Swift)的键名布局存储——卷积层采用NHWC布局,量化线性层按weight/scales/biases三元组存放; - 磁盘上的张量形状、键名、量化格式与运行时内存中的形态一一对应;
- 加载时只需把字节"原样搬进"对应位置,没有任何转换瞬时(zero conversion transient)。
结果就是加载峰值内存 ≈ 常驻内存:在 M5 Max 上实测峰值 14.8GB,没有任何"先膨胀再回落"的内存尖峰。你可以打开 model.safetensors.index.json 验证:每个量化线性层(如self_attn.q_proj)都对应weight、scales、biases三个键——这正是 MLX 量化张量的原生三元组结构。
MoT 双路架构:理解流与生成流
从权重索引还能看出 MoT"双路并行"的设计:每一层 Transformer 都带两套权重——
- 理解流:
mlp、self_attn.q/k/v/o_proj、input_layernorm…(负责 VQA 等理解任务) - 生成流:
mlp_mot_gen、self_attn.*_proj_mot_gen、norm_mot_gen…(负责图像生成,键名带_mot_gen后缀)
8 步蒸馏适配器只触及生成流的 294 个投影,理解流完全不动。这带来两个实测结论(runtime ≥ 0.5.1):
| 能力 | 表现 |
|---|---|
| VQA | 与基础档(base tier)理解流逐 token 一致,答案完全相同 |
| 8 步 cfg-free 编辑 | 与 50 步基础档编辑质量相当(2048² 编辑耗时15s vs 169s,PSNR 28.7dB) |
--think推理模式 | 仅基础档支持,本档不提供 |
三步上手:在 16GB Mac 上快速出图
环境要求:Apple Silicon Mac + Swift 工具链,运行时为sensenova-u1-swift(完整命令与仓库地址见 README.md 的 Quick start 一节)。
# 第 1 步:下载权重(约 11GB,含 tokenizer) hf download mlx-community/SenseNova-U1.5-8B-MoT-8step-4bit --local-dir SenseNova-U1.5-8B-MoT-8step-4bit # 第 2 步:编译 MLX-Swift 运行时 swift build -c release # 第 3 步:一条命令出图(8 步去噪,cfg-free) .build/release/sensenova-cli --weights SenseNova-U1.5-8B-MoT-8step-4bit \ --prompt "A cinematic mountain lake at sunrise, realistic photography." \ --width 1024 --height 1024 --steps 8 --cfg 1.0 --out out.npy参数解读:
--steps 8:8 步蒸馏档的去噪步数,无需调大--cfg 1.0:该档为cfg-free训练,cfg 保持 1.0 即可- 指令编辑加
--edit-image输入参考图;VQA 加--vqa
性能实测与画质权衡
M5 Max 实测(来自 README.md):
| 分辨率 | 耗时 | 峰值内存 |
|---|---|---|
| 1024 × 1024 | 3.2s | 14.8GB |
| 2048 × 2048 | 15.8s | 14.8GB |
需要注意的一个权衡 ⚠️:4bit 量化对扩散去噪轨迹有可测量的扰动——固定 seed 下产出的是"另一张同样合格的图"(与 bf16 参考的余弦相似度约 0.92)。如果你追求与 bf16 逐像素一致的可复现输出,建议选 8-bit 档;追求速度和内存友好,4bit 档是最小的入门层,非常适合作为 Mac 上跑统一多模态模型的第一站。
常见问题
Q1:为什么加载不会内存尖峰?权重在磁盘上就是运行时内存布局(NHWC 卷积 + 量化三元组),加载是"零转换"直读,峰值内存≈常驻的 14.8GB。
Q2:量化后画质损失大吗?生成流上余弦相似度约 0.92(相对 bf16),出图依然可用;但 VQA 走的是未量化的理解流路径,答案与基础档逐 token 一致,编辑质量与 50 步基础档仅差 28.7dB。
Q3:能替代 PyTorch 参考实现吗?运行时与参考 PyTorch 实现的组件级误差 < 1e-4,端到端逐 pass 余弦 0.999+,日常使用完全等价。
Q4:许可证是什么?Apache-2.0,模型归属与致谢 SenseTime / SenseNova 团队,详见 README.md 的 Provenance 一节。
📌小结:SenseNova-U1.5-8B-MoT-8step-4bit 把"8B 统一多模态 + 8 步蒸馏 + 4bit 量化"打包成 Apple Silicon 原生的 MLX 工件,用零转换加载抹平了模型格式转换的内存与时间成本。对于想在 16GB Mac 上体验文生图与图像编辑的用户,这就是目前最轻量的入口之一。
- 人工智能
- 大模型
- 多模态
- 媒体生成
- 模型量化
【免费下载链接】SenseNova-U1.5-8B-MoT-8step-4bit
相关推荐
SenseNova-U1.5-8B-MoT-8step-8bit本地部署实战:sensenova-u1-swift运行时与CLI参数完整指南
SenseNova U1.5 8B MoT 8step 8bit本地部署实战:sensenova u1 swift运行时与CLI参数完整指南 SenseNova
大模型媒体生成多模态本地部署模型量化模型蒸馏理解流与生成流:SenseNova-U1.5-8B-MoT-8step-4bit 的 NEO-unify MoT 双流架构完整拆解
理解流与生成流:SenseNova U1.5 8B MoT 8step 4bit 的 NEO unify MoT 双流架构完整拆解 本文带你读懂 SenseNo
人工智能大模型多模态媒体生成模型量化4bit、8bit还是base档?SenseNova-U1.5-8B-MoT-8step-4bit量化档位选择终极指南
4bit、8bit还是base档?SenseNova U1.5 8B MoT 8step 4bit量化档位选择终极指南 SenseNova U1.5 8B Mo
人工智能大模型多模态媒体生成模型量化
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考