Loading…
Loading…
Two newer releases are available. Haru v2.0 is the current model (17.0M parameters, KoBEST mean 0.469 against a chance mean of 0.450), and Haru v1.1 remains available.
v2.0 is a different architecture and tokenizer, and runs at recurrent depth 6 only, where v1.0 and v1.1 also support depths 2 and 4.
This repository remains available for reproducibility and existing v1.0 users.
Haru is a compact Korean story continuation model built with the custom CFRD causal architecture. It has 6,793,363 parameters and supports recurrent inference depths 2, 4, and 6.
Review the included Python files before enabling remote custom code.
from transformers import AutoModelForCausalLM, AutoTokenizer
MODEL_ID = "alice-noa-chan/haru"
tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained(MODEL_ID, trust_remote_code=True)
inputs = tokenizer("작은 마을에 아침이 찾아왔어요.", return_tensors="pt")
output = model.generate(
**inputs,
max_new_tokens=120,
do_sample=True,
temperature=0.7,
top_p=0.9,
top_k=40,
repetition_penalty=1.08,
use_cache=False,
)
print(tokenizer.decode(output[0], skip_special_tokens=True))
| Recurrent depth | Validation loss | Perplexity |
|---|---|---|
| 2 | 2.37096 | 10.708 |
| 4 | 2.06052 | 7.850 |
| 6 | 2.00630 | 7.436 |
Tiny-Ko-Stories by psymon, licensed under CC BY 4.0. The dataset is not redistributed with this model.
Haru model weights and included code are released under the MIT License. The training dataset remains under its separate CC BY 4.0 license.
Demo Spaces were retired at the owner's request. All model weights and revision history are preserved.