python -m venv openelm_env
source openelm_env/bin/activateOpenELM 官方基于 PyTorch + Transformers。
torch
torchvision
transformers
tokenizers
numpy
safetensors
sentencepiece
huggingface_hub示例(有网时安装):
pip install torch torchvision transformers safetensors sentencepiece⚠️ 离线时必须用 pip download 先下载 wheel 包OpenELM 不使用 tokenizer 自带的权重,需要明确下载:
apple/OpenELM-270Mapple/OpenELM-450Mapple/OpenELM-1_1Bapple/OpenELM-3Bconfig.json
model.safetensors 或 pytorch_model.bin
tokenizer.json
tokenizer_config.json
special_tokens_map.jsonpip download -r requirements.txt -d ./offline_packages
huggingface-cli download apple/OpenELM-1_1B --local-dir ./OpenELM-1_1Boffline_packages/
OpenELM-1_1B/
run_openelm.pypip install --no-index --find-links=./offline_packages *.whlfrom transformers import AutoModelForCausalLM, AutoTokenizer
model = AutoModelForCausalLM.from_pretrained(
"./OpenELM-1_1B",
local_files_only=True
)
tokenizer = AutoTokenizer.from_pretrained(
"./OpenELM-1_1B",
local_files_only=True
)acceleratebitsandbytes(量化)onnxruntime(ONNX 推理)coremltoolsmlx(Mac 上更推荐 MLX 版 OpenELM)❌ 只下载 model.safetensors 不够
❌ transformers 版本过低不支持 OpenELM
如果你愿意,我可以:
你打算用在什么设备上?