54 lines
1.3 KiB
Markdown
54 lines
1.3 KiB
Markdown
---
|
||
language:
|
||
- en
|
||
license: other
|
||
license_name: nvidia-open-model-license
|
||
license_link: >-
|
||
https://developer.download.nvidia.com/licenses/nvidia-open-model-license-agreement-june-2024.pdf
|
||
tags:
|
||
- mlx
|
||
- llm
|
||
- nemotron
|
||
- apple-silicon
|
||
base_model: nvidia/Nemotron-Mini-4B-Instruct
|
||
---
|
||
|
||
# mlx-community/Nemotron-Mini-4B-Instruct-bf16-mlx
|
||
|
||
This model was converted from [nvidia/Nemotron-Mini-4B-Instruct](https://huggingface.co/nvidia/Nemotron-Mini-4B-Instruct)
|
||
to [MLX](https://github.com/ml-explore/mlx) format for use on Apple Silicon.
|
||
|
||
**Quantization:** No quantization – full bfloat16
|
||
|
||
## Usage
|
||
|
||
```python
|
||
from mlx_lm import load, generate
|
||
|
||
model, tokenizer = load("mlx-community/{repo_name}")
|
||
|
||
prompt = (
|
||
"<extra_id_0>System\\n"
|
||
"You are a helpful, honest AI assistant.\\n\\n"
|
||
"<extra_id_1>User\\n"
|
||
"Who are you?\\n"
|
||
"<extra_id_1>Assistant\\n"
|
||
)
|
||
|
||
print(generate(model, tokenizer, prompt, max_tokens=256))
|
||
```
|
||
|
||
## Benchmark (Apple Silicon, single prompt, 23 tokens)
|
||
|
||
| Variant | tok/s |
|
||
|---|---|
|
||
| bf16 (this) | 2.47 |
|
||
| 4-bit default | 4.37 |
|
||
| mxfp4-q4 | 4.56 |
|
||
| nvfp4-q4 | 9.69 |
|
||
| mixed-3-6 | 9.72 |
|
||
|
||
## Original model
|
||
|
||
See [nvidia/Nemotron-Mini-4B-Instruct](https://huggingface.co/nvidia/Nemotron-Mini-4B-Instruct)
|
||
for the original model card, license, and usage terms. |