tinyllama-4bit / README.md
shashikanth-a's picture
85f603b51956efe8ccd6160a7e64809dc3911108e388f12c7c39dba6a991a16a
09d3442 verified
|
raw
history blame
894 Bytes
---
language:
- en
library_name: transformers
license: apache-2.0
tags:
- unsloth
- transformers
- tinyllama
- mlx
base_model: unsloth/tinyllama
---
# shashikanth-a/tinyllama-4bit
The Model [shashikanth-a/tinyllama-4bit](https://huggingface.co/shashikanth-a/tinyllama-4bit) was converted to MLX format from [unsloth/tinyllama](https://huggingface.co/unsloth/tinyllama) using mlx-lm version **0.19.3**.
## Use with mlx
```bash
pip install mlx-lm
```
```python
from mlx_lm import load, generate
model, tokenizer = load("shashikanth-a/tinyllama-4bit")
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)
```