LFM2 1.2B GGUF

🚀 LFM2-1.2B

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

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

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

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

🚀 快速开始

运行模型

要运行 LFM2，你需要从源代码安装 Hugging Face 的 transformers（v4.54.0.dev0）。你可以使用以下命令进行更新或安装：

pip install "transformers @ git+https://github.com/huggingface/transformers.git@main"

以下是一个在 Python 中使用 transformers 生成答案的示例：

from transformers import AutoModelForCausalLM, AutoTokenizer

# 加载模型和分词器
model_id = "LiquidAI/LFM2-1.2B"
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 模型进行微调，以最大化其性能。

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

✨ 主要特性

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

📦 安装指南

运行 LFM2 需要从源代码安装 Hugging Face 的 transformers 库，可使用以下命令：

pip install "transformers @ git+https://github.com/huggingface/transformers.git@main"

💻 使用示例

基础用法

from transformers import AutoModelForCausalLM, AutoTokenizer

# 加载模型和分词器
model_id = "LiquidAI/LFM2-1.2B"
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))

📚 详细文档

模型详情

由于模型规模较小，我们建议在特定用例上对 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 采用了具有乘法门和短卷积的新型混合 Liquid 模型架构，包含 10 个双门短程 LIV 卷积块和 6 个分组查询注意力（GQA）块。

预训练数据

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

训练方法

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

📈 性能

1. 自动化基准测试

2. LLM-as-a-Judge

3. 推理

ExecuTorch 中 CPU 上的吞吐量比较

Llama.cpp 中 CPU 上的吞吐量比较

📄 许可证

本模型使用 LFM 开放许可证 v1.0，具体许可证信息请参考 LICENSE。

📬 联系我们

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

⚠️ 重要提示 包含我们的 聊天模板修复！
对于 llama.cpp，请使用 --jinja

💡 使用建议 由于模型规模较小，建议在特定用例上对 LFM2 模型进行微调，以最大化性能。不建议将其用于知识密集型任务或需要编程技能的任务。