MLX How to use AutomatosX/AX-DeepSeek-V4-Flash-MLX-AXQ-2bit-MTP 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("AutomatosX/AX-DeepSeek-V4-Flash-MLX-AXQ-2bit-MTP")
prompt = "Once upon a time in"
text = generate(model, tokenizer, prompt=prompt, verbose=True) 
