gpt2-xl-bf16 / README.md
warshanks's picture
Upload README.md with huggingface_hub
496e535 verified
---
language: en
license: mit
tags:
- mlx
base_model: openai-community/gpt2-xl
---
# warshanks/gpt2-xl-bf16
The Model [warshanks/gpt2-xl-bf16](https://huggingface.co/warshanks/gpt2-xl-bf16) was converted to MLX format from [openai-community/gpt2-xl](https://huggingface.co/openai-community/gpt2-xl) using mlx-lm version **0.24.1**.
## Use with mlx
```bash
pip install mlx-lm
```
```python
from mlx_lm import load, generate
model, tokenizer = load("warshanks/gpt2-xl-bf16")
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)
```