MLX How to use Aryagm/HRM-Text-1B-MLX-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("Aryagm/HRM-Text-1B-MLX-BF16")
prompt = "Once upon a time in"
text = generate(model, tokenizer, prompt=prompt, verbose=True) 
