MLX How to use N8Programs/ProGen2-base-bf16 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("N8Programs/ProGen2-base-bf16")
prompt = "Once upon a time in"
text = generate(model, tokenizer, prompt=prompt, verbose=True) 
