OpenELM 是 Apple 开源的一系列小参数高效语言模型(270M / 450M / 1.1B / 3B),适合:
| 模型规模 | 最低配置 | 推荐配置 |
|---|---|---|
| 270M / 450M | CPU | 任意笔记本 |
| 1.1B | 8GB RAM | 16GB + GPU |
| 3B | 16GB RAM | 16GB+ 显存(如 RTX 3060/4070) |
pip install torch transformers sentencepiece若有 NVIDIA 显卡,建议安装 CUDA 版本 PyTorch:
pip install torch --index-url https://download.pytorch.org/whl/cu121from transformers import AutoModelForCausalLM, AutoTokenizer
model_id = "apple/OpenELM-1_1B"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(
model_id,
torch_dtype="auto",
device_map="auto"
)
inputs = tokenizer("Hello, who are you?", return_tensors="pt").to("cuda")
outputs = model.generate(**inputs, max_new_tokens=100)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))✅ 优点
git clone https://github.com/apple/corenet
cd corenet
pip install -e .从 Hugging Face 下载:
apple/OpenELM-xxxpython -m corenet.modeling.openelm.generate \
--model apple/OpenELM-1_1B⚠️ 注意
torch not compiled with CUDA解决:
pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121OpenELM 主要训练于英文语料
✅ 建议:
Windows 部署 OpenELM 最稳妥的方式是:Hugging Face Transformers + PyTorch,不碰 CoreNet 官方脚本。
如果你愿意,我可以:
你打算用在什么场景?(本地聊天 / 微调 / 嵌入式)