Add Python to PATHpython --versionpython -m venv openelm_env
openelm_env\Scripts\activatepip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
pip install transformers accelerate如果你有 NVIDIA 显卡,建议安装 CUDA 版本:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118run_openelm.pyfrom transformers import AutoModelForCausalLM, AutoTokenizer
model_name = "apple/OpenELM-270M"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name)
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.pyollama run openelmOllama 会自动下载模型并进入对话模式。
示例:
>>> 你好
你好!有什么我可以帮你的吗?| 模型 | Hugging Face |
|---|---|
| 270M | apple/OpenELM-270M |
| 450M | apple/OpenELM-450M |
| 1.1B | apple/OpenELM-1_1B |
| 3B | apple/OpenELM-3B |
设置镜像:
set HF_ENDPOINT=https://hf-mirror.com| 场景 | 推荐方式 |
|---|---|
| 学习 / 开发 | Python + Transformers |
| 快速体验 | Ollama |
| 无 GPU | CPU + 小模型 |
如果你愿意,我可以帮你:
直接告诉我你的 Windows 版本 + 是否有显卡 即可。