OpenELM 是 Apple 发布的 高效小语言模型系列,特点:
验证:
python --versionpython -m venv openelm_env
openelm_env\Scripts\activatepip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
pip install transformers accelerate如果你 没有 NVIDIA 显卡,用 CPU 版本:
pip install torch torchvision torchaudio创建 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=100)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))运行:
python run_openelm.pypip install huggingface_hub
huggingface-cli download apple/OpenELM-270M --local-dir openelm-270mnvidia-smipip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121model = AutoModelForCausalLM.from_pretrained(
model_id,
device_map="auto"
)DockerfileFROM python:3.10
RUN pip install torch transformers accelerate
WORKDIR /app
COPY run_openelm.py .
CMD ["python", "run_openelm.py"]docker build -t openelm-demo .
docker run --rm openelm-demo✅ 解决方案:
✅ 解决方案:
✅ 推荐:
| 模型 | 适合 |
|---|---|
| OpenELM-270M | 最轻量,CPU 可跑 |
| OpenELM-450M | 平衡 |
| OpenELM-1.1B | 有 GPU 推荐 |
| OpenELM-3B | 高性能机器 |
如果你愿意,我可以:
你打算 本地跑 还是 服务器部署?