How to use smdesai/Nemotron-Labs-Diffusion-3B-4bit with MLX:
# Make sure mlx-lm is installed # pip install --upgrade mlx-lm # Generate text with mlx-lm from mlx_lm import load, generate model, tokenizer = load("smdesai/Nemotron-Labs-Diffusion-3B-4bit") prompt = "Write a story about Einstein" messages = [{"role": "user", "content": prompt}] prompt = tokenizer.apply_chat_template( messages, add_generation_prompt=True ) text = generate(model, tokenizer, prompt=prompt, verbose=True)
How to use smdesai/Nemotron-Labs-Diffusion-3B-4bit with Pi:
# Install MLX LM: uv tool install mlx-lm # Start a local OpenAI-compatible server: mlx_lm.server --model "smdesai/Nemotron-Labs-Diffusion-3B-4bit"
# Install Pi: npm install -g @mariozechner/pi-coding-agent # Add to ~/.pi/agent/models.json: { "providers": { "mlx-lm": { "baseUrl": "http://localhost:8080/v1", "api": "openai-completions", "apiKey": "none", "models": [ { "id": "smdesai/Nemotron-Labs-Diffusion-3B-4bit" } ] } } }
# Start Pi in your project directory: pi
How to use smdesai/Nemotron-Labs-Diffusion-3B-4bit with Hermes Agent:
# Install Hermes: curl -fsSL https://hermes-agent.nousresearch.com/install.sh | bash hermes setup # Point Hermes at the local server: hermes config set model.provider custom hermes config set model.base_url http://127.0.0.1:8080/v1 hermes config set model.default smdesai/Nemotron-Labs-Diffusion-3B-4bit
hermes
How to use smdesai/Nemotron-Labs-Diffusion-3B-4bit with MLX LM:
# Install MLX LM uv tool install mlx-lm # Interactive chat REPL mlx_lm.chat --model "smdesai/Nemotron-Labs-Diffusion-3B-4bit"
# Install MLX LM uv tool install mlx-lm # Start the server mlx_lm.server --model "smdesai/Nemotron-Labs-Diffusion-3B-4bit" # Calling the OpenAI-compatible server with curl curl -X POST "http://localhost:8000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "smdesai/Nemotron-Labs-Diffusion-3B-4bit", "messages": [ {"role": "user", "content": "Hello"} ] }'
