跳到主要内容
抢先遇见新产品的地方

Needle

以单个14MB二进制运行的4500万参数工具调用模型

AI10天前
分类
AI
语言
英语
发布于
10天前
星标
8,764
复刻
566

cactus-compute/needle ↗· Python· Apache-2.0· 最近提交 15小时前

Needle 2是一个面向工具调用、设备操作和结构化抽取的4500万参数开放模型。整个模型就是一个14MB的二进制,一次会话大约占28MB内存。权重被烧进引擎,没有另外需要管理的模型文件,推理过程也不联网。

输入是文本,输出是JSON。由工具schema编译出的字节级语法对每个token强制格式,回复还会附带一个经过校准的置信度。作者说它基于自研的Simple Attention Network成果,用Cactus Quants压到CQ2位;在他们自己的基准上,体积小5到70倍的情况下与FunctionGemma 270M、LFM2.5 230M和Apple FM互有胜负。这个仓库是包含推理、LoRA微调和导出的Python包,用pip install cactus-needle装。

喜欢这个产品吗?