teknium/openhermes
Viewer • Updated • 243k • 545 • 221
How to use marcel/phi-2-openhermes-30k with MLX:
# Make sure mlx-lm is installed
# pip install --upgrade mlx-lm
# if on a CUDA device, also pip install mlx[cuda]
# Generate text with mlx-lm
from mlx_lm import load, generate
model, tokenizer = load("marcel/phi-2-openhermes-30k")
prompt = "Once upon a time in"
text = generate(model, tokenizer, prompt=prompt, verbose=True)How to use marcel/phi-2-openhermes-30k with MLX LM:
# Install MLX LM uv tool install mlx-lm # Generate some text mlx_lm.generate --model "marcel/phi-2-openhermes-30k" --prompt "Once upon a time"
This model was converted to MLX format from microsoft/phi-2.
Refer to the original model card for more details on the model.
pip install mlx
git clone https://github.com/ml-explore/mlx-examples.git
cd mlx-examples/llms/hf_llm
python generate.py --model marcel/phi-2-openhermes-30k --prompt "My name is"
from transformers import AutoModelForCausalLM, AutoTokenizer
model = AutoModelForCausalLM.from_pretrained(
"marcel/phi-2-openhermes-30k",
low_cpu_mem_usage=True,
device_map="auto",
trust_remote_code=True,
torch_dtype=torch.float16,
)
tokenizer = AutoTokenizer.from_pretrained("phi-2-openhermes-30k")
input_text = "### Human: Give me a good recipe for a chinese dish\n\n### Assistant:"
outputs = model.generate(
tokenizer(input_text, return_tensors="pt").to(model.device)['input_ids'],
max_length=1024,
temperature=0.7,
top_p=0.9,
do_sample=True,
pad_token_id=tokenizer.pad_token_id,
eos_token_id=tokenizer.eos_token_id,
)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
Detailed results can be found here
| Metric | Value |
|---|---|
| Avg. | 60.37 |
| AI2 Reasoning Challenge (25-Shot) | 61.01 |
| HellaSwag (10-Shot) | 74.72 |
| MMLU (5-Shot) | 57.17 |
| TruthfulQA (0-shot) | 45.38 |
| Winogrande (5-shot) | 74.90 |
| GSM8k (5-shot) | 49.05 |
Quantized
# Make sure mlx-lm is installed # pip install --upgrade mlx-lm # if on a CUDA device, also pip install mlx[cuda] # Generate text with mlx-lm from mlx_lm import load, generate model, tokenizer = load("marcel/phi-2-openhermes-30k") prompt = "Once upon a time in" text = generate(model, tokenizer, prompt=prompt, verbose=True)