mlx-community

OLMoE-1B-7B-0125-Instruct-4bit

README

License: apache-2.0

Use with mlx

bash

pip install mlx-lm

python

from mlx_lm import load, generate model, tokenizer = load("mlx-community/OLMoE-1B-7B-0125-Instruct-4bit") prompt = "hello" if tokenizer.chat_template is not None:    messages = [{"role": "user", "content": prompt}]    prompt = tokenizer.apply_chat_template(        messages, add_generation_prompt=True    ) response = generate(model, tokenizer, prompt=prompt, verbose=True)