Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3name = "trillionlabs/Tri-0.5B-Base"
4tok = AutoTokenizer.from_pretrained(name)
5model = AutoModelForCausalLM.from_pretrained(
6 name,
7 torch_dtype="bfloat16",
8 device_map="auto"
9)
10
11prompt = "Write a short paragraph about Hangul."
12x = tok(prompt, return_tensors="pt").to(model.device)
13y = model.generate(
14 **x,
15 max_new_tokens=128,
16 do_sample=True,
17 temperature=0.8,
18 top_p=0.95
19)
20print(tok.decode(y[0], skip_special_tokens=True))@misc{trillionlabs_tri05b_base_2025,
title = {Tri-0.5B-Base},
author = {Trillion Labs},
year = {2025},
note = {https://huggingface.co/trillionlabs/Tri-0.5B-Base}
}