Downloads · 30 days
5
1% of all-time downloads
Loke-60000/rin-4b-r1-exp
rin-4b-r1-exp is a text generation model from Loke-60000. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as apache-2.0.
Downloads · 30 days
5
1% of all-time downloads
All-time downloads
701
Public
Parameters
4.2B
8.4 GB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors8.4 GB · 100%
From the Hugging Face model README
from transformers import AutoModelForCausalLM, AutoTokenizer
tok = AutoTokenizer.from_pretrained("Loke-60000/rin-4b-r1-exp")
model = AutoModelForCausalLM.from_pretrained("Loke-60000/rin-4b-r1-exp", device_map="auto")
msgs = [
{"role": "system", "content": "You are Rin. Think privately inside <think>...</think>; after </think>, write the final answer for a human."},
{"role": "user", "content": "If 3 workers build a wall in 12 days, how long for 6?"},
]
ids = tok.apply_chat_template(msgs, add_generation_prompt=True, return_tensors="pt").to(model.device)
print(tok.decode(model.generate(ids, max_new_tokens=512)[0][ids.shape[1]:], skip_special_tokens=True))