How to use mlx-community/gpt2-base-mlx 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/gpt2-base-mlx") prompt = "Once upon a time in" text = generate(model, tokenizer, prompt=prompt, verbose=True)
How to use mlx-community/gpt2-base-mlx with MLX LM:
# Install MLX LM uv tool install mlx-lm # Generate some text mlx_lm.generate --model "mlx-community/gpt2-base-mlx" --prompt "Once upon a time"
