File size: 943 Bytes
f5a6cb9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
---
license: mit
library_name: mlx
language:
- en
base_model: Zyphra/ZR1-1.5B
datasets:
- AI-MO/NuminaMath-CoT
- codeparrot/apps
- deepmind/code_contests
- BAAI/TACO
- MatrixStudio/Codeforces-Python-Submissions
pipeline_tag: text-generation
tags:
- mlx
---
# dinerburger/ZR1-1.5B-mlx-8bit
This model [dinerburger/ZR1-1.5B-mlx-8bit](https://huggingface.co/dinerburger/ZR1-1.5B-mlx-8bit) was
converted to MLX format from [Zyphra/ZR1-1.5B](https://huggingface.co/Zyphra/ZR1-1.5B)
using mlx-lm version **0.22.4**.
## Use with mlx
```bash
pip install mlx-lm
```
```python
from mlx_lm import load, generate
model, tokenizer = load("dinerburger/ZR1-1.5B-mlx-8bit")
prompt = "hello"
if tokenizer.chat_template is not None:
messages = [{"role": "user", "content": prompt}]
prompt = tokenizer.apply_chat_template(
messages, add_generation_prompt=True
)
response = generate(model, tokenizer, prompt=prompt, verbose=True)
```
|