Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3
4base_model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-8b-hf")
5model = PeftModel.from_pretrained(base_model, "cuboniks/tvl-en-llm-translation-stage-b-llama8b")
6tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-8b-hf")
7
8# Inference
9prompt = "Translate to English: [Tuvaluan text]"
10inputs = tokenizer(prompt, return_tensors="pt")
11outputs = model.generate(**inputs)
12print(tokenizer.decode(outputs[0]))1@model{tv2en_translation,
2 title={Stage B: Selective translation with bilingual capability on Llama 2 8B},
3 author={cuboniks},
4 year={2026},
5 publisher={Hugging Face Hub},
6 url={https://huggingface.co/cuboniks/tvl-en-llm-translation-stage-b-llama8b}
7}