tiny / README.md
harpertoken's picture
Upload README.md with huggingface_hub
dd745f6 verified
|
Raw History Blame Contribute Delete
2.19 kB
---
license: apache-2.0
language: en
library_name: mlx
pipeline_tag: text-generation
base_model: mlx-community/SmolLM-135M-Instruct-4bit
tags:
- mlx
- 4-bit
- apple-silicon
- text-generation
---
# tiny
A four-bit quantised conversion of [`mlx-community/SmolLM-135M-Instruct-4bit`](https://huggingface.co/mlx-community/SmolLM-135M-Instruct-4bit), produced with `mlx-lm` 0.30.4 and intended for Apple Silicon. The architecture is a Llama variant (thirty layers, 576 hidden dimensions, nine attention heads, three key-value heads, 49152-token vocabulary) with group-size 64 four-bit weights, occupying about 76 MB.
The `model.safetensors.index.json` alongside the weights is not a sharded checkpoint here; it maps all 694 tensors onto the single `model.safetensors` file that sits beside it.
Because the weights are quantised, `transformers` cannot load this checkpoint; it needs `mlx-lm`, which applies the group quantisation at load time. The instruction tuning lives upstream in SmolLM; this repository adds a format, nothing else. Given the size, expect it to be fast and largely incoherent on factual questions.
## Usage
```python
from mlx_lm import load, generate
model, tokenizer = load("harpertoken/tiny")
prompt = "hello"
if tokenizer.chat_template is not None:
prompt = tokenizer.apply_chat_template(
[{"role": "user", "content": prompt}],
add_generation_prompt=True,
return_dict=False,
)
print(generate(model, tokenizer, prompt=prompt, max_tokens=100))
```
Install with `pip install mlx-lm`. A `chat_template.jinja` is present, so the template branch above will be taken.
## Limitations
This is a redistribution of someone else's work in a different numeric format. There is no training, evaluation or adaptation to report, and if you want SmolLM-135M-Instruct you should use the upstream repository, which is not tied to MLX and can be converted to other formats. Quantisation to four bits costs some accuracy even before the underlying 135M-parameter capacity does; I have not measured the loss and so make no claim about it.
## Attribution
SmolLM follows Allouji et al., *SmolLM: blazingly fast and remarkably powerful* (2024).