深潛:262K長上下文背后的24+8混合注意力設(shè)計(jì))
大模型多模態(tài)模型量化本地部署計(jì)算機(jī)視覺【免費(fèi)下載鏈接】Ternary-Bonsai-2-9B-2bit項(xiàng)目地址https://ai.gitcode.com/hf_mirrors/mlx-community/Ternary-Bonsai-2-9B-2bit點(diǎn)擊查看免費(fèi)下載Ternary-Bonsai-2-9B-2bit是一款專為 Apple Silicon 打造的 9B 級(jí)視覺語言模型采用三元權(quán)重Ternary壓縮至 2-bit 存儲(chǔ)僅占 3.46 GB 磁盤空間卻繼承了 262K 超長上下文能力。其核心是 24 層線性注意力 8 層全局注意力的混合架構(gòu)讓長文本推理在本地設(shè)備上也變得可行且高效。一、模型概覽9B 能力裝進(jìn) 3.46 GB這款模型基于 Qwen3.5-9B 架構(gòu)改造而來語言模型部分約 8.95B 參數(shù)搭配約 0.46B 參數(shù)的官方視覺塔整體以三元值?1、0、1編碼為 MLX 的 2-bit 分組格式組大小 128。相比 FP16 版本約 17.9 GB 的體量磁盤占用壓縮到了約五分之一。規(guī)格項(xiàng)參數(shù)語言模型參數(shù)約 8.95B32 層隱藏維度 4096視覺塔參數(shù)約 0.46B深度 27patch 16權(quán)重格式Ternary {?1, 0, 1}MLX affine 2-bitgroup 128最大上下文262144 tokens262K磁盤占用3.46 GB二、混合注意力24 層線性 8 層全局262K 長上下文能否真正落地關(guān)鍵就在注意力機(jī)制的設(shè)計(jì)上。config.json 中明確列出了 32 層的layer_types分布規(guī)律非常清晰——每 4 層里前 3 層是線性注意力linear_attention第 4 層是全局注意力full_attention由full_attention_interval: 4控制整個(gè) 32 層正好是 8 組循環(huán)線性注意力層24 層linear_num_key_heads: 16、linear_num_value_heads: 32鍵值頭維度各 128卷積核大小 4SSM 狀態(tài)以 float32 計(jì)算。這類層不需要保存完整 KV 緩存顯存占用幾乎不隨上下文增長是長文本能跑上 262K 的根本原因。全局注意力層8 層16 個(gè)查詢頭 4 個(gè)鍵值頭GQA 結(jié)構(gòu)head_dim為 256保留了完整注意力的精確建模能力。這種248的排布是典型的混合注意力Hybrid Attention策略大部分層用廉價(jià)的線性注意力處理長程上下文少量層用全局注意力錨定關(guān)鍵信息兩者互補(bǔ)既省顯存又不失精度。三、三元權(quán)重與 Hadamard 旋轉(zhuǎn)三元權(quán)重并非簡單把浮點(diǎn)數(shù)值四舍五入成三個(gè)檔位而是配合了Hadamard 旋轉(zhuǎn)旋轉(zhuǎn)被折疊進(jìn)了存儲(chǔ)的權(quán)重里運(yùn)行時(shí)再對(duì)激活做匹配的變換。hadamard.json 記錄了變換細(xì)節(jié)——采用歸一化 Sylvester–Walsh–Hadamard塊大小 4096作用于輸入最后一維符號(hào)模式為顯式explicit。這套機(jī)制讓極低的位寬下仍能保持可用的精度。轉(zhuǎn)碼過程由 runtime/codec.py 完成它把 PQ2_0/PTQ1_0 無損地轉(zhuǎn)碼為 MLX 的 affine 2-bit 分塊格式組大小 128runtime/runtime.py 里的fwht函數(shù)則負(fù)責(zé)在推理時(shí)施加前向/逆向 Hadamard 變換。四、MTP 投機(jī)解碼可選加速倉庫還附帶了一個(gè)可選的多 token 預(yù)測頭MTP存放在 mtp/ 目錄以 4-bit、block 2 量化。主模型會(huì)逐 token 校驗(yàn)草稿因此輸出與普通解碼一致。它最適合代碼改寫、重復(fù)性高的文本生成場景能明顯提速但在普通對(duì)話里收益有限默認(rèn)仍走普通解碼。使用方式見 MTP.md。五、快速上手在 Apple Silicon 上跑通非常簡單安裝 mlx-vlm 后加載即可pip install mlx-vlm0.7.8from mlx_vlm import load, generate from mlx_vlm.prompt_utils import apply_chat_template model, processor load(bonsai2-9b-mlx) prompt apply_chat_template(processor, model.config, What is in this image?, num_images1) print(generate(model, processor, prompt, [photo.jpg], max_tokens4096, temperature1.0, top_p0.95, top_k20))采樣參數(shù)建議參考 generation_config.json思考模式用temperature1.0, top_p0.95, top_k20非思考模式可降到temperature0.7, top_p0.80。詳細(xì)加載契約見 PACK-RUNTIME.md一條命令冒煙測試可跑 quickstart.py。小結(jié)Ternary-Bonsai-2-9B-2bit 把大參數(shù)、長上下文、本地運(yùn)行這三件過去難以兼得的事用三元 2-bit 權(quán)重 24/8 混合注意力 Hadamard 旋轉(zhuǎn)組合了起來3.46 GB 的體積、262K 的上下文窗口以及面向 Apple Silicon 原生優(yōu)化的推理路徑讓它成為本地長文本與視覺任務(wù)里一個(gè)相當(dāng)有吸引力的選擇。贊分享大模型多模態(tài)模型量化本地部署計(jì)算機(jī)視覺【免費(fèi)下載鏈接】Ternary-Bonsai-2-9B-2bit項(xiàng)目地址https://ai.gitcode.com/hf_mirrors/mlx-community/Ternary-Bonsai-2-9B-2bit點(diǎn)擊查看免費(fèi)下載相關(guān)推薦Ternary-Bonsai-2-9B架構(gòu)剖析混合注意力與262K超長上下文的秘密Ternary Bonsai 2 9B架構(gòu)剖析混合注意力與262K超長上下文的秘密 本文帶你快速看懂 Ternary Bonsai 2 9B 的架構(gòu)這是 P大模型基礎(chǔ)模型多模態(tài)模型量化本地部署計(jì)算機(jī)視覺2B 小模型憑什么支撐 262K 長上下文Ternary-Bonsai-2-2B-2bit 混合注意力架構(gòu)深度拆解2B 小模型憑什么支撐 262K 長上下文Ternary Bonsai 2 2B 2bit 混合注意力架構(gòu)深度拆解 Ternary Bonsai 2 2B 2人工智能大模型基礎(chǔ)模型多模態(tài)模型量化本地部署計(jì)算機(jī)視覺如何微調(diào)InternVL3-1B-AWQ基于SWIFT與XTuner的多模態(tài)模型定制指南如何微調(diào)InternVL3 1B AWQ基于SWIFT與XTuner的多模態(tài)模型定制指南 InternVL3 1B AWQ 是 OpenGVLab 發(fā)布的 I上一篇MHY_Scanner游戲登錄自動(dòng)化革命告別手動(dòng)掃碼時(shí)代下一篇WVP-PRO視頻監(jiān)控平臺(tái)實(shí)戰(zhàn)指南從零構(gòu)建企業(yè)級(jí)安防系統(tǒng)創(chuàng)作聲明:本文部分內(nèi)容由AI輔助生成(AIGC),僅供參考