python3 -m venv openelm-env
source openelm-env/bin/activateOpenELM 位于 CoreNet 仓库中:
git clone https://github.com/apple/corenet.git
cd corenetCPU 版本
pip install torch torchvisionGPU 版本(示例 CUDA 12.1)
pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121pip install -r requirements.txt
pip install -e .OpenELM 模型在 Hugging Face:
pip install huggingface_hub示例(270M 模型):
huggingface-cli download apple/OpenELM-270Mpython projects/openelm/main.py \
model=openelm_270m \
dataset=text_imagenet \
mode=evalfrom transformers import AutoModelForCausalLM, AutoTokenizer
model_id = "apple/OpenELM-270M"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(model_id)
inputs = tokenizer("Hello, OpenELM!", return_tensors="pt")
outputs = model.generate(**inputs, max_length=50)
print(tokenizer.decode(outputs[0]))projects/openelm/README.mdpython projects/openelm/main.py \
model=openelm_270m \
dataset=your_dataset \
optimizer=adamw \
mode=train可以,但只适合推理和小模型。
OpenELM 使用 逐层缩放,参数效率更高。
bitsandbytes 或 gguf 工具。projects/openelm/如果你需要:
可以直接告诉我你的使用场景。