huangang's picture
Upload README.md with huggingface_hub
490e6e9 verified
---
base_model: katanemo/Arch-Router-1.5B
language:
- en
library_name: transformers
license: other
license_name: katanemo-research
license_link: https://huggingface.co/katanemo/Arch-Router-1.5B/blob/main/LICENSE
pipeline_tag: text-generation
tags:
- routing
- preference
- arxiv:2506.16655
- llm
- mlx
- mlx-my-repo
paper: https://arxiv.org/abs/2506.16655
---
# huangang/Arch-Router-1.5B-mlx-fp16
The Model [huangang/Arch-Router-1.5B-mlx-fp16](https://huggingface.co/huangang/Arch-Router-1.5B-mlx-fp16) was converted to MLX format from [katanemo/Arch-Router-1.5B](https://huggingface.co/katanemo/Arch-Router-1.5B) using mlx-lm version **0.22.3**.
## Use with mlx
```bash
pip install mlx-lm
```
```python
from mlx_lm import load, generate
model, tokenizer = load("huangang/Arch-Router-1.5B-mlx-fp16")
prompt="hello"
if hasattr(tokenizer, "apply_chat_template") and tokenizer.chat_template is not None:
messages = [{"role": "user", "content": prompt}]
prompt = tokenizer.apply_chat_template(
messages, tokenize=False, add_generation_prompt=True
)
response = generate(model, tokenizer, prompt=prompt, verbose=True)
```