【免费下载链接】Ornith-1.5-35B-A3B-8bit
Ornith-1.5-35B-A3B-8bit 是一款面向 Apple Silicon 的MoE(Mixture of Experts,混合专家)多模态大模型。它拥有 35B 总参数,却靠"256 个专家里只激活 8 个"的路由策略,让每个 token 实际只计算约 3B 参数(即 A3B),是本地部署大模型的优选方案。本文用通俗的方式,带你读懂它的MoE 路由与稀疏激活机制。
先搞懂:什么是 MoE 路由?🔍
传统大模型每生成一个字,都要动用全部参数,算力开销巨大。MoE 换了个思路:把模型里的前馈网络(MLP)拆成一大群"专家",每个 token 只挑选其中几个最合适的专家来干活,其余专家直接"休眠"。
这就是"稀疏激活"的核心——参数很多,但真正参与计算的很少。
类比:一家医院有 256 位专科医生,患者(token)来了,分诊台(路由器)只叫最相关的 8 位医生会诊,不会让 256 位医生全部出场。
Ornith-1.5-35B-A3B-8bit 正是这种设计:
- 256 个专家:每层都有 256 个候选专家(
num_experts: 256) - 只激活 8 个:每个 token 只路由给 8 个专家(
num_experts_per_tok: 8) - 共享专家兜底:另有一个"共享专家"始终参与,保证基础能力
这些数字都定义在 config.json 的text_config中,可直接查阅。
路由器(Router)是怎么"选"出 8 个专家的?🎯
每一步推理,路由器都要做一次"快速决策":
- 打分:路由器是一个轻量线性层,输入当前 token 的隐状态(维度 2048),对 256 个专家各算一个"匹配分数"(logit)。
- 选 Top-8:从 256 个分数里挑出最高的 8 个专家。
- 加权融合:8 个专家各自输出结果,再按分数加权求和,得到这一层的最终表示。
在权重文件里,这部分叫gate(路由器权重,形状约为 256×2048),而 256 个专家本身被打包在switch_mlp模块中高效存储。你可以在 model.safetensors.index.json 中搜到mlp.gate、mlp.shared_expert、mlp.switch_mlp等键名。
这里还有个细节:配置里router_aux_loss_coef: 0.0,意味着推理阶段不启用辅助负载均衡损失——路由器更"自由"地按内容分配专家,把负载均衡交给训练阶段处理。
共享专家:为什么还要一个"常驻医生"?👨⚕️
除了 8 个被选中的路由专家,模型还配了一个共享专家(shared_expert),它对每个 token 都参与计算,负责沉淀最通用、最基础的语言能力,避免"专家太专精、共性能力被稀释"。
更妙的是,它还带一个独立的门控shared_expert_gate:一个可学习的小线性层,动态决定共享专家输出"贡献多少"。相当于给常驻医生配了一个"音量旋钮",让模型自己判断该让通用知识发挥多大作用。
每个 token 经过某一层时: ├─ 8 个路由专家(Top-8 选出,各自不同) ├─ 1 个共享专家(永远激活) └─ shared_expert_gate(调节共享专家的权重) ──────────────── 加权求和 → 该层最终输出A3B:30 亿激活参数到底从哪来?⚡
名字里的"35B-A3B"是它的招牌:
- 35B= 模型总参数约 350 亿
- A3B= 每个 tokenActive(激活)约 3B(30 亿)
为什么能"省"这么多?因为一次只动 8/256 个专家。粗算一下每层的"活跃"部分:
| 组成 | 每层专家数 | 单个专家参数量(约) | 说明 |
|---|---|---|---|
| 路由专家 | 8 | 3×2048×512 ≈ 3.1M | moe_intermediate_size: 512 |
| 共享专家 | 1 | ≈ 3.1M | 始终激活 |
| 路由器 + 门控 | — | 约 0.5M | 计算开销很小 |
仅 MoE 部分,每层约激活 29M 参数,乘以 40 层(num_hidden_layers: 40)就是十几亿,再叠加注意力层与嵌入层,合计落在约 3B量级——这正是"A3B"的由来。
💡 换句话说:总参数 35B,实际每字只算 ~3B,算力消耗约为同规模稠密模型的 1/10 左右,速度自然快、显存压力也小。
40 层混合注意力:线性 + 全注意力交替排布📊
除了 MoE,这个模型还用了混合注意力来进一步提速:
- 线性注意力(linear_attention):计算更省,适合长文本
- 全注意力(full_attention):表达力更强,用于关键位置
配置full_attention_interval: 4定义了节奏——每 4 层插入 1 层全注意力,其余 3 层用线性注意力。40 层的排布在 config.json 的layer_types里清晰可见。
此外:
- 16 个注意力头 + 2 个 KV 头(GQA 分组查询注意力),进一步压缩缓存
- 支持 256K 上下文(
max_position_embeddings: 262144),可处理超长文档 - 248320 词表(
vocab_size),覆盖多语言
8-bit 量化:让它在 Mac 上跑得动的关键🍎
这个版本是8-bit 量化版,专为 Apple Silicon 的MLX框架优化:
- 量化方式:
bits: 8、group_size: 64、mode: affine(每 64 个权重一组,用一组缩放因子还原) - 文件大小:共 8 个分片(
model-00001-of-00008.safetensors~model-00008-of-00008.safetensors),合计约35.13 GiB
量化把每个参数从 16 位压到 8 位,体积近乎减半,让 35B 级别的模型能塞进消费级 Mac 的统一内存里运行。每个权重复带.weight(量化值)、.scales(缩放)、.biases(偏移)三件套,正是 affine 量化的标准形态。
多模态能力:不止于文字👁️
从configuration.json(task: visual-question-answering)可以看出,它天生支持图像 + 视频理解:
- 视觉塔:27 层(
vision_config.depth: 27),支持 16×16 图像块切分 - 图像 / 视频 token:
image_token_id、video_token_id已在 config.json 中定义 - 对话模板:chat_template.jinja 专门处理图文混合内容的渲染
也就是说,你不仅能让它聊天、写代码,还能把图片或视频丢给它做视觉问答。
如何在本地部署运行?🚀
Ornith-1.5-35B-A3B-8bit 是标准MLX 格式模型,天然适配 Apple Silicon。最简流程:
- 获取模型(在 macOS + Apple Silicon 上执行):
git clone https://gitcode.com/hf_mirrors/mlx-community/Ornith-1.5-35B-A3B-8bit- 确认硬件:模型约 35 GiB,建议配备36GB 及以上统一内存的 Mac 以获得流畅体验。
- 推理生成:用 MLX 生态的
mlx_lm加载目录即可跑起来,推荐参数参考 generation_config.json(temperature: 1.0、top_k: 20、top_p: 0.95)。
关键配置文件速览📁
| 文件 | 作用 |
|---|---|
| config.json | 核心架构:MoE 专家数、路由、量化、注意力排布 |
| generation_config.json | 默认采样参数(温度、top_k/top_p) |
| configuration.json | 框架与任务类型(多模态视觉问答) |
| model.safetensors.index.json | 全部 2090 个张量的分布索引 |
| chat_template.jinja | 图文对话渲染模板 |
| tokenizer.json / tokenizer_config.json | 分词器与配置 |
| preprocessor_config.json / video_preprocessor_config.json | 图像 / 视频预处理 |
model-0000*.safetensors(×8) | 8-bit 量化权重分片 |
一句话总结✅
Ornith-1.5-35B-A3B-8bit 用256 选 8 的 MoE 路由+常驻共享专家,把 35B 的"大个子"压缩成每字约 3B 的"轻快身板",再叠加混合注意力与8-bit 量化,最终实现了"在 Mac 上也能流畅跑大模型"的平衡点。读懂它的路由与激活机制,你也就掌握了当下主流大模型"大而快"的核心设计哲学。
【免费下载链接】Ornith-1.5-35B-A3B-8bit
相关推荐
35B总参只需3B激活?拆解Ornith-1.5-35B-A3B-OptiQ-4bit的MoE混合专家架构
35B总参只需3B激活?拆解Ornith 1.5 35B A3B OptiQ 4bit的MoE混合专家架构 Ornith 1.5 35B A3B OptiQ 4
人工智能大模型基础模型多模态本地部署模型量化128个专家只激活3B:NVIDIA-Nemotron-3.5-Lightning-30B-A3B-8bit的MoE路由与共享专家机制完全解析
128个专家只激活3B:NVIDIA Nemotron 3.5 Lightning 30B A3B 8bit的MoE路由与共享专家机制完全解析 NVIDIA N
256个专家只激活8个:Hy4-preview-4bit的MoE路由与负载均衡机制详解
256个专家只激活8个:Hy4 preview 4bit的MoE路由与负载均衡机制详解 Hy4 preview 4bit 是腾讯 Hy4 preview MoE
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考