Text Generation
Transformers
Safetensors
MLX
llama
code
Eval Results (legacy)
text-generation-inference
Instructions to use mlx-community/granite-3b-code-base-8bit with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use mlx-community/granite-3b-code-base-8bit with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="mlx-community/granite-3b-code-base-8bit")# Load model directly from transformers import AutoTokenizer, AutoModelForCausalLM tokenizer = AutoTokenizer.from_pretrained("mlx-community/granite-3b-code-base-8bit") model = AutoModelForCausalLM.from_pretrained("mlx-community/granite-3b-code-base-8bit") - MLX
How to use mlx-community/granite-3b-code-base-8bit 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("mlx-community/granite-3b-code-base-8bit") prompt = "Once upon a time in" text = generate(model, tokenizer, prompt=prompt, verbose=True) - Notebooks
- Google Colab
- Kaggle
- Local Apps
- LM Studio
- vLLM
How to use mlx-community/granite-3b-code-base-8bit with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "mlx-community/granite-3b-code-base-8bit" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "mlx-community/granite-3b-code-base-8bit", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }'Use Docker
docker model run hf.co/mlx-community/granite-3b-code-base-8bit
- SGLang
How to use mlx-community/granite-3b-code-base-8bit with SGLang:
Install from pip and serve model
# Install SGLang from pip: pip install sglang # Start the SGLang server: python3 -m sglang.launch_server \ --model-path "mlx-community/granite-3b-code-base-8bit" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "mlx-community/granite-3b-code-base-8bit", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }'Use Docker images
docker run --gpus all \ --shm-size 32g \ -p 30000:30000 \ -v ~/.cache/huggingface:/root/.cache/huggingface \ --env "HF_TOKEN=<secret>" \ --ipc=host \ lmsysorg/sglang:latest \ python3 -m sglang.launch_server \ --model-path "mlx-community/granite-3b-code-base-8bit" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "mlx-community/granite-3b-code-base-8bit", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }' - MLX LM
How to use mlx-community/granite-3b-code-base-8bit with MLX LM:
Generate or start a chat session
# Install MLX LM uv tool install mlx-lm # Generate some text mlx_lm.generate --model "mlx-community/granite-3b-code-base-8bit" --prompt "Once upon a time"
- Docker Model Runner
How to use mlx-community/granite-3b-code-base-8bit with Docker Model Runner:
docker model run hf.co/mlx-community/granite-3b-code-base-8bit
| license: apache-2.0 | |
| library_name: transformers | |
| tags: | |
| - code | |
| - mlx | |
| datasets: | |
| - codeparrot/github-code-clean | |
| - bigcode/starcoderdata | |
| - open-web-math/open-web-math | |
| - math-ai/StackMathQA | |
| metrics: | |
| - code_eval | |
| pipeline_tag: text-generation | |
| inference: false | |
| model-index: | |
| - name: granite-3b-code-base | |
| results: | |
| - task: | |
| type: text-generation | |
| dataset: | |
| name: MBPP | |
| type: mbpp | |
| metrics: | |
| - type: pass@1 | |
| value: 36.0 | |
| name: pass@1 | |
| - task: | |
| type: text-generation | |
| dataset: | |
| name: MBPP+ | |
| type: evalplus/mbppplus | |
| metrics: | |
| - type: pass@1 | |
| value: 45.1 | |
| name: pass@1 | |
| - task: | |
| type: text-generation | |
| dataset: | |
| name: HumanEvalSynthesis(Python) | |
| type: bigcode/humanevalpack | |
| metrics: | |
| - type: pass@1 | |
| value: 36.6 | |
| name: pass@1 | |
| - type: pass@1 | |
| value: 37.2 | |
| name: pass@1 | |
| - type: pass@1 | |
| value: 40.9 | |
| name: pass@1 | |
| - type: pass@1 | |
| value: 26.2 | |
| name: pass@1 | |
| - type: pass@1 | |
| value: 35.4 | |
| name: pass@1 | |
| - type: pass@1 | |
| value: 22.0 | |
| name: pass@1 | |
| - type: pass@1 | |
| value: 25.0 | |
| name: pass@1 | |
| - type: pass@1 | |
| value: 18.9 | |
| name: pass@1 | |
| - type: pass@1 | |
| value: 29.9 | |
| name: pass@1 | |
| - type: pass@1 | |
| value: 17.1 | |
| name: pass@1 | |
| - type: pass@1 | |
| value: 26.8 | |
| name: pass@1 | |
| - type: pass@1 | |
| value: 14.0 | |
| name: pass@1 | |
| - type: pass@1 | |
| value: 18.3 | |
| name: pass@1 | |
| - type: pass@1 | |
| value: 23.2 | |
| name: pass@1 | |
| - type: pass@1 | |
| value: 29.9 | |
| name: pass@1 | |
| - type: pass@1 | |
| value: 24.4 | |
| name: pass@1 | |
| - type: pass@1 | |
| value: 16.5 | |
| name: pass@1 | |
| - type: pass@1 | |
| value: 3.7 | |
| name: pass@1 | |
| # mlx-community/granite-3b-code-base-8bit | |
| The Model [mlx-community/granite-3b-code-base-8bit](https://huggingface.co/mlx-community/granite-3b-code-base-8bit) was converted to MLX format from [ibm-granite/granite-3b-code-base](https://huggingface.co/ibm-granite/granite-3b-code-base) using mlx-lm version **0.12.0**. | |
| ## Use with mlx | |
| ```bash | |
| pip install mlx-lm | |
| ``` | |
| ```python | |
| from mlx_lm import load, generate | |
| model, tokenizer = load("mlx-community/granite-3b-code-base-8bit") | |
| response = generate(model, tokenizer, prompt="hello", verbose=True) | |
| ``` | |