推荐 Python 3.10 或 3.11
验证:
python --versionOpenELM 已上传到 Hugging Face:
https://huggingface.co/apple/OpenELM常用模型:
OpenELM-270MOpenELM-450MOpenELM-1_1BOpenELM-3Bgit clone https://github.com/apple/corenet.git
cd corenetpython -m venv openelm_env
openelm_env\Scripts\activatepip install torch torchvision torchaudiopip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118验证:
import torch
print(torch.cuda.is_available())pip install transformers accelerate创建 run_openelm.py:
from transformers import AutoModelForCausalLM, AutoTokenizer
model_name = "apple/OpenELM-270M" # 可换成 450M / 1_1B / 3B
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(
model_name,
torch_dtype="auto",
device_map="auto"
)
prompt = "Once upon a time"
inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
outputs = model.generate(
**inputs,
max_new_tokens=100,
do_sample=True,
temperature=0.8
)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))运行:
python run_openelm.py✅ 优点:
解决:
nvidia-smi| 场景 | 推荐方案 |
|---|---|
| 新手 | LM Studio |
| 开发者 | Python + Transformers |
| 低配置 | OpenELM-270M |
| 中等配置 | 1.1B |
| 高配置 GPU | 3B |
如果你需要:
告诉我你的 用途 + 硬件配置,我可以直接给你定制方案。