LFM2 700M

🚀 LFM2-700M

LFM2 是由 Liquid AI 开发的新一代混合模型，专为边缘 AI 和设备端部署而设计。它在质量、速度和内存效率方面树立了新的标准。

我们发布了三个经过后续训练的检查点的权重，分别具有 3.5 亿、7 亿和 12 亿参数。它们为创建 AI 驱动的边缘应用程序提供了以下关键特性：

• 快速训练和推理：与上一代相比，LFM2 的训练速度提高了 3 倍。与 Qwen3 相比，它在 CPU 上的解码和预填充速度也快了 2 倍。
• 卓越性能：在多个基准测试类别中，包括知识、数学、指令遵循和多语言能力，LFM2 的性能优于同等规模的模型。
• 全新架构：LFM2 是一种新型混合 Liquid 模型，具有乘法门和短卷积。
• 灵活部署：LFM2 可在 CPU、GPU 和 NPU 硬件上高效运行，便于在智能手机、笔记本电脑或车辆上灵活部署。

在我们的 博客文章 中了解更多关于 LFM2 的信息。

📦 安装指南

你可以使用 transformers 和 llama.cpp 运行 LFM2，vLLM 支持即将推出。

1. Transformers

要运行 LFM2，你需要从源代码（v4.54.0.dev0）安装 Hugging Face 的 transformers。 你可以使用以下命令更新或安装它：pip install "transformers @ git+https://github.com/huggingface/transformers.git@main"。

2. Llama.cpp

你可以使用 GGUF 检查点 通过 llama.cpp 运行 LFM2。在模型卡片中查找更多信息。

💻 使用示例

基础用法

from transformers import AutoModelForCausalLM, AutoTokenizer

# 加载模型和分词器
model_id = "LiquidAI/LFM2-700M"
model = AutoModelForCausalLM.from_pretrained(
    model_id,
    device_map="auto",
    torch_dtype="bfloat16",
    trust_remote_code=True,
#    attn_implementation="flash_attention_2" <- 在兼容的 GPU 上取消注释
)
tokenizer = AutoTokenizer.from_pretrained(model_id)

# 生成答案
prompt = "What is C. elegans?"
input_ids = tokenizer.apply_chat_template(
    [{"role": "user", "content": prompt}],
    add_generation_prompt=True,
    return_tensors="pt",
    tokenize=True,
).to(model.device)

output = model.generate(
    input_ids,
    do_sample=True,
    temperature=0.3,
    min_p=0.15,
    repetition_penalty=1.05,
    max_new_tokens=512,
)

print(tokenizer.decode(output[0], skip_special_tokens=False))

# <|startoftext|><|im_start|>user
# What is C. elegans?<|im_end|>
# <|im_start|>assistant
# C. elegans, also known as Caenorhabditis elegans, is a small, free-living
# nematode worm (roundworm) that belongs to the phylum Nematoda.

你可以使用这个 Colab 笔记本 直接运行和测试模型。

📚 详细文档

📄 模型详情

由于其规模较小，我们建议在特定用例上对 LFM2 模型进行微调，以充分发挥其性能。它们特别适用于代理任务、数据提取、RAG、创意写作和多轮对话。但是，我们不建议将它们用于知识密集型或需要编程技能的任务。

支持的语言：英语、阿拉伯语、中文、法语、德语、日语、韩语和西班牙语。

生成参数：我们建议使用以下参数：

• temperature=0.3
• min_p=0.15
• repetition_penalty=1.05

聊天模板：LFM2 使用类似 ChatML 的聊天模板，如下所示：

<|startoftext|><|im_start|>system
You are a helpful assistant trained by Liquid AI.<|im_end|>
<|im_start|>user
What is C. elegans?<|im_end|>
<|im_start|>assistant
It's a tiny nematode that lives in temperate soil environments.<|im_end|>

你可以使用 Hugging Face transformers 中的专用 .apply_chat_template() 函数来应用它。

工具使用：它包括四个主要步骤：

1. 函数定义：LFM2 将 JSON 函数定义作为输入（<|tool_list_start|> 和 <|tool_list_end|> 特殊标记之间的 JSON 对象），通常在系统提示中。
2. 函数调用：LFM2 编写 Python 风格的函数调用（<|tool_call_start|> 和 <|tool_call_end|> 特殊标记之间的 Python 列表），作为助手的回答。
3. 函数执行：执行函数调用并返回结果（<|tool_response_start|> 和 <|tool_response_end|> 特殊标记之间的字符串），作为“工具”角色。
4. 最终答案：LFM2 解释函数调用的结果，以纯文本形式回答原始用户提示。

以下是一个使用工具的简单对话示例：

<|startoftext|><|im_start|>system
List of tools: <|tool_list_start|>[{"name": "get_candidate_status", "description": "Retrieves the current status of a candidate in the recruitment process", "parameters": {"type": "object", "properties": {"candidate_id": {"type": "string", "description": "Unique identifier for the candidate"}}, "required": ["candidate_id"]}}]<|tool_list_end|><|im_end|>
<|im_start|>user
What is the current status of candidate ID 12345?<|im_end|>
<|im_start|>assistant
<|tool_call_start|>[get_candidate_status(candidate_id="12345")]<|tool_call_end|>Checking the current status of candidate ID 12345.<|im_end|>
<|im_start|>tool
<|tool_response_start|>{"candidate_id": "12345", "status": "Interview Scheduled", "position": "Clinical Research Associate", "date": "2023-11-20"}<|tool_response_end|><|im_end|>
<|im_start|>assistant
The candidate with ID 12345 is currently in the "Interview Scheduled" stage for the position of Clinical Research Associate, with an interview date set for 2023-11-20.<|im_end|>

架构：具有乘法门和短卷积的混合模型：10 个双门短程 LIV 卷积块和 6 个分组查询注意力（GQA）块。

预训练数据混合：大约 75% 的英语数据、20% 的多语言数据和 5% 的代码数据，来源于网络和许可材料。

训练方法：

• 使用 LFM1-7B 作为教师模型进行知识蒸馏。
• 在 50% 的下游任务和 50% 的通用领域上进行大规模的监督微调（SFT）。
• 使用长度归一化和半在线数据集的自定义直接偏好优化（DPO）。
• 迭代模型合并。

🔧 如何微调 LFM2

我们建议在你的用例上对 LFM2 模型进行微调，以充分发挥其性能。

笔记本	描述	链接
SFT（Axolotl）	在 Axolotl 中使用 LoRA 适配器进行监督微调（SFT）的笔记本。
SFT（TRL）	在 TRL 中使用 LoRA 适配器进行监督微调（SFT）的笔记本。
DPO（TRL）	在 TRL 中使用直接偏好优化（DPO）进行偏好对齐。

📈 性能

LFM2 在不同的评估类别中优于同等规模的模型。

1. 自动化基准测试

2. LLM 裁判评估

3. 推理

CPU 上的吞吐量比较（ExecuTorch）

CPU 上的吞吐量比较（Llama.cpp）

📄 许可证

本项目使用 LFM 开放许可证 v1.0，详情见 LICENSE。

📬 联系我们

如果您对边缘部署的定制解决方案感兴趣，请联系 我们的销售团队。