OpenELM 是 Apple 开源的一系列高效率 LLM,特点:
projects/openelm)推荐 Python 3.10
python --version如未安装:
CPU 版
pip install torch torchvision torchaudioGPU(CUDA 11.8)
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118git clone https://github.com/apple/corenet
cd corenet
pip install -e .pip install huggingface_hub示例(OpenELM-3B-Instruct):
from huggingface_hub import snapshot_download
snapshot_download(
repo_id="apple/OpenELM-3B-Instruct",
local_dir="models/openelm-3b-instruct"
)其他模型:
apple/OpenELM-270Mapple/OpenELM-450M-Instructapple/OpenELM-1_1B-Chatfrom transformers import AutoModelForCausalLM, AutoTokenizer
model_path = "models/openelm-3b-instruct"
tokenizer = AutoTokenizer.from_pretrained(model_path)
model = AutoModelForCausalLM.from_pretrained(
model_path,
torch_dtype="auto",
device_map="auto"
)
prompt = "请介绍一下 OpenELM 模型。"
inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
output = model.generate(
**inputs,
max_new_tokens=200
)
print(tokenizer.decode(output[0], skip_special_tokens=True))"cuda" 改为 "cpu"git clone https://github.com/oobabooga/text-generation-webui
cd text-generation-webui
pip install -r requirements.txtmodels/ 目录python server.py⚠️ OpenELM 非原生支持,需要:
适合:
pip install -e .corenet 根目录| 场景 | 推荐方案 |
|---|---|
| 学习 / 实验 | Python + HuggingFace |
| 本地聊天 | Text Generation WebUI |
| 低配机器 | llama.cpp(转换后) |
| 研究训练 | CoreNet 官方流程 |
如果你愿意,我可以:
你打算用在什么场景?