bigcode/commitpack
Updated • 58.8k • 79
How to use mlx-community/Mellum-4b-sft-python-4bit 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("mlx-community/Mellum-4b-sft-python-4bit")
prompt = "Once upon a time in"
text = generate(model, tokenizer, prompt=prompt, verbose=True)How to use mlx-community/Mellum-4b-sft-python-4bit with MLX LM:
# Install MLX LM uv tool install mlx-lm # Generate some text mlx_lm.generate --model "mlx-community/Mellum-4b-sft-python-4bit" --prompt "Once upon a time"
This model mlx-community/Mellum-4b-sft-python-4bit was converted to MLX format from JetBrains/Mellum-4b-sft-python using mlx-lm version 0.25.2.
pip install mlx-lm
from mlx_lm import load, generate
model, tokenizer = load("mlx-community/Mellum-4b-sft-python-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)
4-bit
Base model
JetBrains/Mellum-4b-base