Views
No views yet
pip install -U transformers1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4model_id = "ai-for-good-lab/byol-nya-4b-it"
5tokenizer = AutoTokenizer.from_pretrained(model_id)
6model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", dtype=torch.bfloat16)
7
8# Chat inference
9messages = [{"role": "user", "content": "Tandiuzeni za dziko la Malawi."}]
10inputs = tokenizer.apply_chat_template(messages, return_tensors="pt", add_generation_prompt=True, return_dict=True).to(model.device)
11outputs = model.generate(**inputs, max_new_tokens=256)
12print(tokenizer.decode(outputs[0], skip_special_tokens=True))1@article{zamir2026byolbringlanguagellms,
2 title={BYOL: Bring Your Own Language Into LLMs},
3 author={Syed Waqas Zamir and Wassim Hamidouche and Boulbaba Ben Amor and Luana Marotti and Inbal Becker-Reshef and Juan Lavista Ferres},
4 year={2026},
5 journal={arXiv:2601.10804},
6 url={https://arxiv.org/abs/2601.10804},
7}