Text Generation
MLX
Safetensors
Dream
code
text-diffusion-model
diffusion large language model
conversational
custom_code
8-bit precision
Instructions to use mlx-community/DiffuCoder-7B-cpGRPO-8bit with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- MLX
How to use mlx-community/DiffuCoder-7B-cpGRPO-8bit 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("mlx-community/DiffuCoder-7B-cpGRPO-8bit") 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) - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- LM Studio
- Pi
How to use mlx-community/DiffuCoder-7B-cpGRPO-8bit with Pi:
Start the MLX server
# Install MLX LM: uv tool install mlx-lm # Start a local OpenAI-compatible server: mlx_lm.server --model "mlx-community/DiffuCoder-7B-cpGRPO-8bit"
Configure the model in Pi
# 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": "mlx-community/DiffuCoder-7B-cpGRPO-8bit" } ] } } }Run Pi
# Start Pi in your project directory: pi
- Hermes Agent new
How to use mlx-community/DiffuCoder-7B-cpGRPO-8bit with Hermes Agent:
Start the MLX server
# Install MLX LM: uv tool install mlx-lm # Start a local OpenAI-compatible server: mlx_lm.server --model "mlx-community/DiffuCoder-7B-cpGRPO-8bit"
Configure Hermes
# 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 mlx-community/DiffuCoder-7B-cpGRPO-8bit
Run Hermes
hermes
- MLX LM
How to use mlx-community/DiffuCoder-7B-cpGRPO-8bit with MLX LM:
Generate or start a chat session
# Install MLX LM uv tool install mlx-lm # Interactive chat REPL mlx_lm.chat --model "mlx-community/DiffuCoder-7B-cpGRPO-8bit"
Run an OpenAI-compatible server
# Install MLX LM uv tool install mlx-lm # Start the server mlx_lm.server --model "mlx-community/DiffuCoder-7B-cpGRPO-8bit" # Calling the OpenAI-compatible server with curl curl -X POST "http://localhost:8000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "mlx-community/DiffuCoder-7B-cpGRPO-8bit", "messages": [ {"role": "user", "content": "Hello"} ] }'
issues when running the provided script for inference
#2
by atinsood - opened
ERROR:root:Model type Dream not supported.
Traceback (most recent call last):
File "/home/opensource/ml-diffucoder/.venv/lib/python3.11/site-packages/mlx_lm/utils.py", line 67, in _get_classes
arch = importlib.import_module(f"mlx_lm.models.{model_type}")
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/Users/boomer/Library/Application Support/uv/python/cpython-3.11.13-macos-aarch64-none/lib/python3.11/importlib/__init__.py", line 126, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "<frozen importlib._bootstrap>", line 1204, in _gcd_import
File "<frozen importlib._bootstrap>", line 1176, in _find_and_load
File "<frozen importlib._bootstrap>", line 1140, in _find_and_load_unlocked
ModuleNotFoundError: No module named 'mlx_lm.models.Dream'
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File "/home/opensource/ml-diffucoder/inference_demo_mac.py", line 3, in <module>
model, tokenizer = load("mlx-community/DiffuCoder-7B-cpGRPO-8bit")
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/opensource/ml-diffucoder/.venv/lib/python3.11/site-packages/mlx_lm/utils.py", line 241, in load
model, config = load_model(model_path, lazy)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/opensource/ml-diffucoder/.venv/lib/python3.11/site-packages/mlx_lm/utils.py", line 176, in load_model
model_class, model_args_class = get_model_classes(config=config)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/opensource/ml-diffucoder/.venv/lib/python3.11/site-packages/mlx_lm/utils.py", line 71, in _get_classes
raise ValueError(msg)
ValueError: Model type Dream not supported.
version of mlx-lm being used
mlx-lm==0.26.3