验证:
python --version打开 CMD / PowerShell,执行:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
pip install transformers accelerate sentencepiece如果有 NVIDIA 显卡,可装 CUDA 版 PyTorch(更快)
新建 run_openelm.py:
from transformers import AutoModelForCausalLM, AutoTokenizer
model_id = "apple/OpenELM-270M"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(model_id)
prompt = "Once upon a time"
inputs = tokenizer(prompt, return_tensors="pt")
outputs = model.generate(**inputs, max_length=50)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))运行:
python run_openelm.py| 模型 | 说明 |
|---|---|
apple/OpenELM-450M | 平衡 |
apple/OpenELM-1_1B | 效果好 |
apple/OpenELM-3B | 需要好 CPU / GPU |
model_id 即可。设置镜像:
set HF_ENDPOINT=https://hf-mirror.compip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121可以接 Gradio:
pip install gradio✅ Windows
✅ 不用编译
✅ 5 分钟跑通
✅ 官方模型直接可用
如果你愿意,我可以:
你想走到哪一步?