OpenELM 官方示例大量使用:
bashcoreutilsargparse + torchWindows 直接跑脚本容易出问题,建议:
推荐:
Python 3.10 / 3.11如果你是 GPU 部署(推荐):
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121如果你用的是:
ollama ❌ 官方暂不支持 OpenELMllama.cpp ❌ 需转换,不推荐 Windows 新手✅ 推荐方式:
transformers 直接加载OpenELM 需要:
pip install transformers>=4.40示例(Windows 可用):
from transformers import AutoModelForCausalLM, AutoTokenizer
model = AutoModelForCausalLM.from_pretrained(
"apple/OpenELM-270M",
torch_dtype="auto",
device_map="auto"
)
tokenizer = AutoTokenizer.from_pretrained("apple/OpenELM-270M")⚠️ Windows 下 device_map="auto" 有时不稳定,建议:
model.to("cuda")避免:
"C:\models\openelm"✅ 用:
r"C:\models\openelm"或
"C:/models/openelm"❌ 不要放:
C:\用户\桌面\Open ELM\model✅ 放:
D:\models\openelm| 模型 | 显存 | 备注 |
|---|---|---|
| 270M | 2GB | 可 CPU |
| 450M | 3GB | 入门 GPU |
| 1.1B | 6GB | 3060 可用 |
| 3B | 12GB+ | 建议 4070 / 3090 |
❌ 不要指望:
✅ 最稳方案:
最快方式(Windows):
pip install openelm如果你愿意,我可以:
你打算 GPU 还是 CPU?模型多大?