Model: mlx-community/sarashina2.1-1b-sft Source: Original Platform
library_name, license, license_link, base_model, tags, language, model-index
| library_name | license | license_link | base_model | tags | language | model-index | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| transformers | other | LICENSE | Aratako/sarashina2.1-1b-sft |
|
|
|
mlx-community/sarashina2.1-1b-sft
The Model mlx-community/sarashina2.1-1b-sft was converted to MLX format from Aratako/sarashina2.1-1b-sft using mlx-lm version 0.20.2.
Use with mlx
pip install mlx-lm
from mlx_lm import load, generate
model, tokenizer = load("mlx-community/sarashina2.1-1b-sft")
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)
Description